Finalize tests

luxonis · Jan 13, 2024 · 7327cbf · 7327cbf
1 parent 41e1211
commit 7327cbf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/tests/integration/test_pipeline.py b/tests/integration/test_pipeline.py
@@ -221,8 +221,8 @@ def test_cuda_simple_sdxl_turbo_detection_pipeline():
 
 
 @pytest.mark.skipif(
-    total_memory < 12 or total_disk_space < 16,
-    reason="Test requires at least 12GB of RAM and 16GB of HDD",
+    total_memory < 16 or total_disk_space < 16,
+    reason="Test requires at least 16GB of RAM and 16GB of HDD",
 )
 def test_cpu_simple_sdxl_detection_pipeline():
     # Define target folder
@@ -243,8 +243,8 @@ def test_cpu_simple_sdxl_detection_pipeline():
 
 
 @pytest.mark.skipif(
-    not torch.cuda.is_available() or total_memory < 12 or total_disk_space < 16,
-    reason="Test requires GPU, at least 12GB of RAM and 16GB of HDD",
+    not torch.cuda.is_available() or total_memory < 16 or total_disk_space < 16,
+    reason="Test requires GPU, at least 16GB of RAM and 16GB of HDD",
 )
 def test_cuda_simple_sdxl_detection_pipeline():
     # Define target folder
@@ -396,8 +396,8 @@ def test_cuda_simple_sdxl_turbo_classification_pipeline():
 
 
 @pytest.mark.skipif(
-    total_memory < 12 or total_disk_space < 16,
-    reason="Test requires at least 12GB of RAM and 16GB of HDD",
+    total_memory < 16 or total_disk_space < 16,
+    reason="Test requires at least 16GB of RAM and 16GB of HDD",
 )
 def test_cpu_simple_sdxl_classification_pipeline():
     # Define target folder
@@ -419,8 +419,8 @@ def test_cpu_simple_sdxl_classification_pipeline():
 
 
 @pytest.mark.skipif(
-    not torch.cuda.is_available() or total_memory < 12 or total_disk_space < 16,
-    reason="Test requires GPU, at least 12GB of RAM and 16GB of HDD",
+    not torch.cuda.is_available() or total_memory < 16 or total_disk_space < 16,
+    reason="Test requires GPU, at least 16GB of RAM and 16GB of HDD",
 )
 def test_cuda_simple_sdxl_classification_pipeline():
     # Define target folder

diff --git a/tests/unittests/test_image_generation.py b/tests/unittests/test_image_generation.py
@@ -72,16 +72,16 @@ def test_cpu_sdxl_image_generator():
 
 
 @pytest.mark.skipif(
-    not torch.cuda.is_available() or total_memory < 12 or total_disk_space < 16,
-    reason="Test requires GPU,at least 12GB of RAM and 16GB of HDD",
+    not torch.cuda.is_available() or total_memory < 16 or total_disk_space < 16,
+    reason="Test requires GPU, at least 16GB of RAM and 16GB of HDD",
 )
 def test_cuda_sdxl_turbo_image_generator():
     _check_image_generator(StableDiffusionTurboImageGenerator, "cuda")
 
 
 @pytest.mark.skipif(
-    total_memory < 12 or total_disk_space < 16,
-    reason="Test requiresat least 12GB of RAM and 16GB of HDD",
+    total_memory < 16 or total_disk_space < 16,
+    reason="Test requires at least 16GB of RAM and 16GB of HDD",
 )
 def test_cpu_sdxl_turbo_image_generator():
     _check_image_generator(StableDiffusionTurboImageGenerator, "cpu")