Test suite

bedogni@unimore.it · bedogni@unimore.it · commit bfd3452f8b69 · 2026-04-14T17:49:41.000+02:00
diff --git a/tests/fixtures/mqtt_client_fixture.py b/tests/fixtures/mqtt_client_fixture.py
@@ -11,31 +11,30 @@
 
 @pytest.fixture
 def offloading_data_fixture(monkeypatch):
-    """ Fixture to override OffloadingDataFiles paths for all tests. """
+    """Fixture to override OffloadingDataFiles paths for all tests."""
     base_path = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
-    monkeypatch.setattr(OffloadingDataFiles, "data_file_path_device",
-                        os.path.join(base_path, TestSamples.data_file_path_device))
-    monkeypatch.setattr(OffloadingDataFiles, "data_file_path_edge",
-                        os.path.join(base_path, TestSamples.data_file_path_edge))
-    monkeypatch.setattr(OffloadingDataFiles, "data_file_path_sizes",
-                        os.path.join(base_path, TestSamples.data_file_path_sizes))
+    monkeypatch.setattr(
+        OffloadingDataFiles,
+        "data_file_path_sizes",
+        os.path.join(base_path, TestSamples.data_file_path_sizes),
+    )
 
     return OffloadingDataFiles  # Optional: return for reference if needed
 
 
 @pytest.fixture
 def mqtt_client_fixture(offloading_data_fixture):
-    """ Fixture to create an MQTT client with overridden file paths. """
-    broker_url = 'hostname.local'
+    """Fixture to create an MQTT client with overridden file paths."""
+    broker_url = "hostname.local"
     broker_port = 1883
-    client_id = 'edge'
+    client_id = "edge"
     topics = {
-      'registration': 'devices/',
-      'offloading_layer': 'device_01/offloading_layer',
-      'device_input': 'device_01/input_data',
-      'device_inference_result': 'device_01/model_inference_result'
+        "registration": "devices/",
+        "offloading_layer": "device_01/offloading_layer",
+        "device_input": "device_01/input_data",
+        "device_inference_result": "device_01/model_inference_result",
     }
-    ntp_server = '0.it.pool.ntp.org'
+    ntp_server = "0.it.pool.ntp.org"
     input_height = 96
     input_width = 96
     last_offloading_layer = 58
@@ -49,22 +48,22 @@ def mqtt_client_fixture(offloading_data_fixture):
         input_height=input_height,
         input_width=input_width,
         last_offloading_layer=last_offloading_layer,
-        request_handler=RequestHandler()
+        request_handler=RequestHandler(),
     )
 
 
 @pytest.fixture
 def device_fixture(mqtt_client_fixture):
-    broker_url = 'hostname.local'
+    broker_url = "hostname.local"
     broker_port = 1883
-    client_id = 'edge'
+    client_id = "edge"
     topics = {
-      'registration': 'devices/',
-      'offloading_layer': 'device_01/offloading_layer',
-      'device_input': 'device_01/input_data',
-      'device_inference_result': 'device_01/model_inference_result'
+        "registration": "devices/",
+        "offloading_layer": "device_01/offloading_layer",
+        "device_input": "device_01/input_data",
+        "device_inference_result": "device_01/model_inference_result",
     }
-    ntp_server = '0.it.pool.ntp.org'
+    ntp_server = "0.it.pool.ntp.org"
     input_height = 96
     input_width = 96
     last_offloading_layer = 58
@@ -78,5 +77,5 @@ def device_fixture(mqtt_client_fixture):
         input_height=input_height,
         input_width=input_width,
         last_offloading_layer=last_offloading_layer,
-        request_handler=RequestHandler()
+        request_handler=RequestHandler(),
     )
diff --git a/tests/test_edge/test_edge.py b/tests/test_edge/test_edge.py
@@ -1,7 +1,24 @@
 from server.edge.edge_initialization import Edge
 
+
 def test_edge():
     e = Edge()
     height = 96
     width = 96
-    e.initialization(input_height=height, input_width=width)
+    model_dir = "test_model_96x96"
+    model_key = "fomo_96x96"
+    models_config = {
+        "fomo_96x96": {
+            "input_height": 96,
+            "input_width": 96,
+            "last_offloading_layer": 58,
+            "model_dir": "test_model_96x96",
+        }
+    }
+    e.initialization(
+        input_height=height,
+        input_width=width,
+        model_dir=model_dir,
+        model_key=model_key,
+        models_config=models_config,
+    )
