test added

soeren-kirchner · Oct 16, 2020 · e14ac26 · e14ac26
1 parent 98611ac
commit e14ac26
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/pyimgbatch/test_pyimgbatch.py b/pyimgbatch/test_pyimgbatch.py
@@ -0,0 +1,17 @@
+import unittest
+from pyimgbatch import to_int_or_none
+
+
+class TestToIntOrNone(unittest.TestCase):
+
+    def test_to_int_or_none(self):
+        result = to_int_or_none(5)
+        self.assertIsNotNone(result)
+        self.assertEqual(result, 5)
+        result = to_int_or_none("5")
+        self.assertIsNotNone(result)
+        self.assertEqual(result, 6)
+
+
+if __name__ == "__main__":
+    unittest.main()