filter ast definition and validation

1 files changed, 16 insertions, 1 deletions

diff --git a/test/utils/test_commons.py b/test/utils/test_commons.py
index ce73788..3ad6dea 100644
--- a/test/utils/test_commons.py
+++ b/test/utils/test_commons.py
@@ -8,7 +8,7 @@ Author: Matthias Baumgartner, 2022
 import unittest
 
 # objects to test
-from bsfs.utils.commons import typename
+from bsfs.utils.commons import typename, normalize_args
 
 
 ## code ##
@@ -21,6 +21,21 @@ class TestCommons(unittest.TestCase):
         self.assertEqual(typename(123), 'int')
         self.assertEqual(typename(None), 'NoneType')
 
+    def test_normalize_args(self):
+        # one argument
+        self.assertEqual(normalize_args(1), (1, ))
+        # pass as arguments
+        self.assertEqual(normalize_args(1,2,3), (1,2,3))
+        # pass as iterator
+        self.assertEqual(normalize_args(iter([1,2,3])), (1,2,3))
+        # pass as generator
+        self.assertEqual(normalize_args((i for i in range(1, 4))), (1,2,3))
+        self.assertEqual(normalize_args(i for i in range(1, 4)), (1,2,3)) # w/o brackets
+        # pass as iterable
+        self.assertEqual(normalize_args([1,2,3]), (1,2,3))
+        # pass an iterable with a single item
+        self.assertEqual(normalize_args([1]), (1, ))
+
 
 
 ## main ##