Update test_pdastack.py

caleb531 · Oct 21, 2023 · 7f38399 · 7f38399
1 parent 44331d2
commit 7f38399
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_pdastack.py b/tests/test_pdastack.py
@@ -28,3 +28,13 @@ def test_stack_get(self) -> None:
     def test_stack_repr(self) -> None:
         """Should create proper string representation of PDA stack."""
         self.assertEqual(repr(self.stack), "PDAStack(('a', 'b'))")
+
+    def test_stack_equality(self) -> None:
+        """Should only be equal for equal configurations."""
+        stack = PDAStack(["a", "b"])
+
+        self.assertEqual(stack, stack)
+        self.assertEqual(stack, PDAStack(["a", "b"]))
+
+        self.assertNotEqual(stack, PDAStack(["a", "a"]))
+        self.assertNotEqual(stack, PDAStack(["b", "b"]))