tests update

tldr-group · Jul 8, 2024 · 873a3c9 · 873a3c9
1 parent 801a5e8
commit 873a3c9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -5,3 +5,4 @@ matplotlib==3.8.1
 scikit-image==0.22.0
 scikit-learn==1.3.1
 tifffile==2024.7.2
+flake8==6.1.0
diff --git a/tests/tests.py b/tests/tests.py
@@ -101,7 +101,7 @@ def test_cls_disk(self):
         print(
             f"CLS={integral_range:.3f}, VF={vf:.3f} for {n_disks} disks with diameter {2 * radius} on {x}x{y} image \n"
         )
-        plt.imsave("disk.png", arr)
+        # plt.imsave("disk.png", arr)
         assert np.isclose(integral_range, 2 * radius, rtol=0.05)
 
     def test_cls_squares(self):
@@ -120,7 +120,7 @@ def test_cls_squares(self):
 
         tpc = model.radial_tpc(arr, False, False)
         integral_range = model.tpc_to_cls(tpc, arr)
-        plt.imsave("foo.png", arr)
+        # plt.imsave("foo.png", arr)
         print(integral_range, np.sqrt(2) * l)
 
     def test_repr_pred(self):