Remove the large test files

RapidAI · Feb 4, 2024 · fb3886a · fb3886a
1 parent a2e11df
commit fb3886a
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/python/tests/test_files/two_dim_image.npy b/python/tests/test_files/two_dim_image.npy
diff --git a/python/tests/test_ort.py b/python/tests/test_ort.py
@@ -141,7 +141,8 @@ def test_input_three_ndim_two_channel():
     image_array = np.load(str(img_npy))
     result, _ = engine(image_array)
 
-    assert len(result) == 87
+    assert len(result) == 1
+    assert result[0][1] == " TREND PLOT REPORT"
 
 
 def test_input_three_ndim_one_channel():

diff --git a/python/tests/test_paddle.py b/python/tests/test_paddle.py
@@ -141,7 +141,8 @@ def test_input_three_ndim_two_channel():
     image_array = np.load(str(img_npy))
     result, _ = engine(image_array)
 
-    assert len(result) == 87
+    assert len(result) == 1
+    assert result[0][1] == " TREND PLOT REPORT"
 
 
 def test_input_three_ndim_one_channel():

diff --git a/python/tests/test_vino.py b/python/tests/test_vino.py
@@ -141,7 +141,8 @@ def test_input_three_ndim_two_channel():
     image_array = np.load(str(img_npy))
     result, _ = engine(image_array)
 
-    assert len(result) == 87
+    assert len(result) == 1
+    assert result[0][1] == " TREND PLOT REPORT"
 
 
 def test_input_three_ndim_one_channel():