Add deocrator to skip specific HW related unit tests on CI

AlexMontgomerie · Feb 8, 2024 · 325bf3d · 325bf3d
1 parent e4d13b8
commit 325bf3d
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 10 deletions.
diff --git a/tests/test_buffer_depth_estimation.py b/tests/test_buffer_depth_estimation.py
@@ -17,6 +17,7 @@
 
 ABS_TOL = 4000
 
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt()
 def test_unet_single_branch_network():
 
@@ -57,7 +58,7 @@ def test_unet_single_branch_network():
     assert net.partitions[0].graph.nodes["Concat_11"]["hw"].buffer_depth[1] == pytest.approx(2720, abs=ABS_TOL)
 
 
-
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt()
 def test_unet_two_branch_network():
 

diff --git a/tests/test_layers.py b/tests/test_layers.py
@@ -357,7 +357,7 @@ def test_layer_configurations(self, config_path):
         self.run_test_wait_depth(layer)
         self.run_test_resources(layer)
 
-
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 class TestConvolutionLayer_HW(TestLayerTemplate,unittest.TestCase):
 
@@ -432,6 +432,7 @@ def test_layer_configurations(self, test_folder_path):
         assert modeling_latency == pytest.approx(simulation_latency, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling latency: {modeling_latency}, simulation latency: {simulation_latency}"
         assert modeling_pipeline_depth == pytest.approx(simulation_pipeline_depth, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling pipeline depth: {modeling_pipeline_depth}, simulation pipeline depth: {simulation_pipeline_depth}"
 
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 class TestPoolingLayer_HW(TestLayerTemplate,unittest.TestCase):
 
@@ -495,7 +496,7 @@ def test_layer_configurations(self, test_folder_path):
         assert modeling_latency == pytest.approx(simulation_latency, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling latency: {modeling_latency}, simulation latency: {simulation_latency}"
         assert modeling_pipeline_depth == pytest.approx(simulation_pipeline_depth, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling pipeline depth: {modeling_pipeline_depth}, simulation pipeline depth: {simulation_pipeline_depth}"
 
-
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 class TestGlobalPoolingLayer_HW(TestLayerTemplate,unittest.TestCase):
 

diff --git a/tests/test_modules.py b/tests/test_modules.py
@@ -49,7 +49,6 @@ def load_module_configs_db(name):
     return list(configs)
 
 
-print(load_module_configs_db("accum"))
 
 class TestModuleTemplate():
 
@@ -276,6 +275,7 @@ def test_module_configurations(self, config_path):
         self.run_test_resources(module)
 
 
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 @pytest.mark.skip(reason="Not implemented yet")
 class TestPadModule_HW(TestModuleTemplate,unittest.TestCase):
@@ -335,7 +335,7 @@ def test_module_configurations(self, test_folder_path):
         assert modeling_latency == pytest.approx(simulation_latency, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling latency: {modeling_latency}, simulation latency: {simulation_latency}"
         assert modeling_pipeline_depth == pytest.approx(simulation_pipeline_depth, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling pipeline depth: {modeling_pipeline_depth}, simulation pipeline depth: {simulation_pipeline_depth}"
 
-
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 @pytest.mark.skip(reason="Not implemented yet")
 class TestSlidingWindowModule_HW(TestModuleTemplate,unittest.TestCase):
@@ -397,7 +397,7 @@ def test_module_configurations(self, test_folder_path):
         assert modeling_latency == pytest.approx(simulation_latency, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling latency: {modeling_latency}, simulation latency: {simulation_latency}"
         assert modeling_pipeline_depth == pytest.approx(simulation_pipeline_depth, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling pipeline depth: {modeling_pipeline_depth}, simulation pipeline depth: {simulation_pipeline_depth}"
 
-
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 @pytest.mark.skip(reason="Not implemented yet")
 class TestAccumModule_HW(TestModuleTemplate,unittest.TestCase):
@@ -455,7 +455,7 @@ def test_module_configurations(self, test_folder_path):
         assert modeling_latency == pytest.approx(simulation_latency, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling latency: {modeling_latency}, simulation latency: {simulation_latency}"
         assert modeling_pipeline_depth == pytest.approx(simulation_pipeline_depth, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling pipeline depth: {modeling_pipeline_depth}, simulation pipeline depth: {simulation_pipeline_depth}"
 
-
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 @pytest.mark.skip(reason="Not implemented yet")
 class TestSqueezeModule_HW(TestModuleTemplate,unittest.TestCase):
@@ -513,7 +513,7 @@ def test_module_configurations(self, test_folder_path):
         assert modeling_latency == pytest.approx(simulation_latency, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling latency: {modeling_latency}, simulation latency: {simulation_latency}"
         assert modeling_pipeline_depth == pytest.approx(simulation_pipeline_depth, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling pipeline depth: {modeling_pipeline_depth}, simulation pipeline depth: {simulation_pipeline_depth}"
 
-
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 @pytest.mark.skip(reason="Not implemented yet")
 class TestVectorDotModule_HW(TestModuleTemplate,unittest.TestCase):
@@ -571,7 +571,7 @@ def test_module_configurations(self, test_folder_path):
         assert modeling_latency == pytest.approx(simulation_latency, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling latency: {modeling_latency}, simulation latency: {simulation_latency}"
         assert modeling_pipeline_depth == pytest.approx(simulation_pipeline_depth, abs=ABS_TOL, rel=REL_TOL), f"TEST {test_id}: Modeling pipeline depth: {modeling_pipeline_depth}, simulation pipeline depth: {simulation_pipeline_depth}"
 
-
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 @pytest.mark.skip(reason="Not implemented yet")
 class TestBiasModule_HW(TestModuleTemplate,unittest.TestCase):

diff --git a/tests/test_pipeline_depth_estimation.py b/tests/test_pipeline_depth_estimation.py
@@ -21,7 +21,7 @@
 def filter_by_type(layer_type):
     return "squeeze" not in layer_type.lower() and "split" not in layer_type.lower() and "reshape" not in layer_type.lower()
 
-@pytest.mark.skip(reason="Currently disabled.")
+@unittest.skip("Currently disabled in CI.")
 @ddt.ddt
 class TestPipelineDepth(unittest.TestCase):