Add tests for MOVE.

test_wordle.py

@@ -15,7 +15,7 @@ class TestConstraints(unittest.TestCase):
         self.assertTrue(constraints.matches('abcde'))
         self.assertTrue(constraints.matches('bcdef'))
 
-    def test_correct_single(self):
+    def test_correct(self):
         constraints = wordle.Constraints()
         constraints.update_graded('abcde', [LetterResult.CORRECT, LetterResult.INCORRECT, LetterResult.INCORRECT, LetterResult.INCORRECT, LetterResult.INCORRECT])
 
@@ -27,7 +27,7 @@ class TestConstraints(unittest.TestCase):
         self.assertFalse(constraints.matches('aedcb'))
         self.assertFalse(constraints.matches('amnob'))
 
-    def test_correct_duplicate_letter(self):
+    def test_correct_count_limit(self):
         constraints = wordle.Constraints()
         constraints.update_graded('aazzz', [LetterResult.CORRECT, LetterResult.INCORRECT, LetterResult.INCORRECT, LetterResult.INCORRECT, LetterResult.INCORRECT])
 
@@ -40,7 +40,17 @@ class TestConstraints(unittest.TestCase):
         self.assertFalse(constraints.matches('bbbba'))
         self.assertFalse(constraints.matches('babbb'))
 
+    def test_move(self):
+        constraints = wordle.Constraints()
+        constraints.update_graded('abcde', [LetterResult.MOVE, LetterResult.INCORRECT, LetterResult.INCORRECT, LetterResult.INCORRECT, LetterResult.INCORRECT])
 
+        self.assertTrue(constraints.matches('zzzza'))
+        self.assertTrue(constraints.matches('zzzaa'))
+        self.assertTrue(constraints.matches('zaaaa'))
+
+        self.assertFalse(constraints.matches('azzzz'))
+        self.assertFalse(constraints.matches('zzzzz'))
+        self.assertFalse(constraints.matches('zzzba'))
 
 
 if __name__ == '__main__':