added tests

test_pmpv.py

@@ -0,0 +1,162 @@
+import unittest
+from io import StringIO
+from unittest.mock import patch
+from pmpv import Variables, evaluate_tokens, tokenize, main
+
+class TestVariables(unittest.TestCase):
+    def setUp(self):
+        self.variables = Variables.get_instance()
+
+    def tearDown(self):
+        self.variables.clear()
+
+    def test_get_set(self):
+        self.variables.set('name', 'John')
+        self.assertEqual(self.variables.get('name'), 'John')
+
+    def test_get_nonexistent(self):
+        self.assertIsNone(self.variables.get('age'))
+
+    def test_contains(self):
+        self.variables.set('name', 'John')
+        self.assertTrue(self.variables.contains('name'))
+        self.assertFalse(self.variables.contains('age'))
+
+    def test_clear(self):
+        self.variables.set('name', 'John')
+        self.variables.clear()
+        self.assertIsNone(self.variables.get('name'))
+
+    def test_str_representation(self):
+        self.variables.set('name', 'John')
+        self.assertEqual(str(self.variables), "{'name': 'John'}")
+
+    def test_repr_representation(self):
+        self.variables.set('name', 'John')
+        self.assertEqual(repr(self.variables), "{'name': 'John'}")
+
+class TestTokenize(unittest.TestCase):
+    def test_valid_input(self):
+        userInput = "x = 5 + (4 - 2)"
+        expected_tokens = ['x', '=', 5, '+', '(', 4, '-', 2, ')']
+        self.assertEqual(tokenize(userInput), expected_tokens)
+
+    def test_invalid_parentheses(self):
+        userInput = "x = (5 + 2"
+        self.assertIsNone(tokenize(userInput))
+
+    def test_invalid_assignment(self):
+        userInput = "x = 5 = 10"
+        self.assertIsNone(tokenize(userInput))
+
+    def test_variable_not_defined(self):
+        userInput = "x = y"
+        self.assertIsNone(tokenize(userInput))
+
+    def test_convert_numbers(self):
+        userInput = "x = 10 + 5.5"
+        expected_tokens = ['x', '=', 10, '+', 5, 5]
+        self.assertEqual(tokenize(userInput), expected_tokens)
+
+class TestEvaluateTokens(unittest.TestCase):
+    def setUp(self):
+        self.variables = Variables.get_instance()
+
+    def tearDown(self):
+        self.variables.clear()
+
+    def test_empty_expression(self):
+        tokens = []
+        result = evaluate_tokens(tokens)
+        self.assertIsNone(result)
+
+    def test_single_value_expression(self):
+        tokens = [5]
+        result = evaluate_tokens(tokens)
+        self.assertEqual(result, 5)
+
+    def test_variable_assignment(self):
+        tokens = ['x', '=', 5]
+        result = evaluate_tokens(tokens)
+        self.assertIsNone(result)
+        self.assertEqual(self.variables.get('x'), 5)
+
+    def test_invalid_syntax(self):
+        tokens = ['-', '+']
+        result = evaluate_tokens(tokens)
+        self.assertIsNone(result)
+
+    def test_parenthesized_expression(self):
+        tokens = ['(', 5, '+', 2, ')']
+        result = evaluate_tokens(tokens)
+        self.assertEqual(result, 7)
+
+    def test_invalid_parentheses(self):
+        tokens = ['(', 5, '+', 2]
+        result = evaluate_tokens(tokens)
+        self.assertIsNone(result)
+
+    def test_invalid_assignment(self):
+        tokens = ['x', '=', 5, '=', 10]
+        result = evaluate_tokens(tokens)
+        self.assertIsNone(result)
+
+    def test_invalid_token(self):
+        tokens = ['x', '=', 'y']
+        result = evaluate_tokens(tokens)
+        self.assertIsNone(result)
+
+    def test_valid_expression(self):
+        tokens = ['x', '=', 5, '+', '(', 4, '-', 2, ')']
+        result = evaluate_tokens(tokens)
+        self.assertIsNone(result)
+        self.assertEqual(self.variables.get('x'), 7)
+
+    def test_invalid_operator(self):
+        tokens = [5, '*', 2]
+        result = evaluate_tokens(tokens)
+        self.assertIsNone(result)
+
+class TestMain(unittest.TestCase):
+    def setUp(self):
+        self.variables = Variables.get_instance()
+
+    def tearDown(self):
+        self.variables.clear()
+
+    def test_integration(self):
+        inputs = [
+            "3 + 5 - -2 - 2",
+            "x = 3 + 5 - -2 - 2",
+            "y = x - (x - 2)",
+            "y",
+            "ans = (17 - (5 - 20)) - (1 - 11)",
+            "ans",
+            "",
+            "ans = ans - (-42 - ans)",
+            "ans",
+            "(17-(5-20))-(1-11)"
+        ]
+        expected_outputs = [
+            "8",
+            "",
+            "",
+            "2",
+            "",
+            "42",
+            "",
+            "126",
+            "42"
+        ]
+
+        with patch('builtins.input', side_effect=inputs), patch('sys.stdout', new=StringIO()) as fake_output:
+            main()
+
+            output_lines = fake_output.getvalue().strip().split('\n')
+            self.assertEqual(len(output_lines), len(expected_outputs))
+
+            for i in range(len(output_lines)):
+                self.assertEqual(output_lines[i], expected_outputs[i])
+
+if __name__ == '__main__':
+    unittest.main()