From 873a3c9d4e37fcb011178d19d380b9ec2e0bad3c Mon Sep 17 00:00:00 2001
From: rmdocherty <rmdocherty2@gmai.com>
Date: Mon, 8 Jul 2024 15:51:57 +0100
Subject: [PATCH] tests update

---
 requirements.txt | 1 +
 tests/tests.py   | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 1b370af..a181d00 100644
--- 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
\ No newline at end of file
diff --git a/tests/tests.py b/tests/tests.py
index 17bf749..45afdbe 100644
--- 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):