diff --git a/test_wordle.py b/test_wordle.py index b58703b..d3a8e4e 100644 --- a/test_wordle.py +++ b/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__':