Added unit tests for folded argument classes.

Author: MarcCote

smartdispatch/tests/test_argument.py

@@ -0,0 +1,39 @@
+from nose.tools import assert_equal, assert_true
+from numpy.testing import assert_array_equal
+
+from smartdispatch.argument import EnumerationFoldedArgument, RangeFoldedArgument
+import smartdispatch
+
+import re
+import os
+import unittest
+
+
+def test_enumeration_folded_argument():
+    # Test valid enumeration folded arguments
+    arg = EnumerationFoldedArgument()
+    folded_arguments = [("[]", [""]),
+                        ("[1]", ["1"]),
+                        ("[1 ]", ["1", ""]),
+                        ("[1 2 3]", ["1", "2", "3"]),
+                        ]
+
+    for folded_argument, unfolded_arguments in folded_arguments:
+        match = re.match(arg.regex, folded_argument)
+        assert_true(match is not None)
+        assert_array_equal(arg.unfold(match.group()), unfolded_arguments)
+
+    # Test invalid enumeration folded arguments
+    assert_true(re.match(arg.regex, "[1 2 3") is None)
+
+
+def test_range_folded_argument():
+    arg = RangeFoldedArgument()
+    folded_arguments = [("[1:4]", ["1", "2", "3"]),
+                        ("[1:4:2]", ["1", "3"]),
+                        ]
+
+    for folded_argument, unfolded_arguments in folded_arguments:
+        match = re.match(arg.regex, folded_argument)
+        assert_true(match is not None)
+        assert_array_equal(arg.unfold(match.group()), unfolded_arguments)