From acfd993bae71971ea4b4f4a2dd0e7804a9a7da70 Mon Sep 17 00:00:00 2001
From: Richard Brown <33289025+rijobro@users.noreply.github.com>
Date: Mon, 16 Aug 2021 11:30:37 +0100
Subject: [PATCH] add unittest.main

Signed-off-by: Richard Brown <33289025+rijobro@users.noreply.github.com>
---
 tests/test_savitzky_golay_filter.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/test_savitzky_golay_filter.py b/tests/test_savitzky_golay_filter.py
index 9163204810..c9bcd9687e 100644
--- a/tests/test_savitzky_golay_filter.py
+++ b/tests/test_savitzky_golay_filter.py
@@ -150,3 +150,7 @@ class TestSavitzkyGolayGPUREP(unittest.TestCase):
     def test_value(self, arguments, image, expected_data, atol):
         result = SavitzkyGolayFilter(**arguments)(image.to(device="cuda"))
         np.testing.assert_allclose(result.cpu(), expected_data, atol=atol)
+
+
+if __name__ == "__main__":
+    unittest.main()