diff --git a/tests/tests.py b/tests/tests.py index 7489c0d..b4b5a42 100644 --- a/tests/tests.py +++ b/tests/tests.py @@ -1328,6 +1328,43 @@ class TestPromptPostProcessor(TestPromptPostProcessorBase): ppp=self.nocupppp, ) + # Variable-vs-variable comparison tests + + def test_cmd_if_var_vs_var_eq(self): # var eq var: both set to same value, if-branch taken + self.process( + PromptPair( + "hellohelloYESNO", + "", + ), + PromptPair("YES", ""), + ) + + def test_cmd_if_var_vs_var_ne(self): # var ne var: different values, ne condition true + self.process( + PromptPair( + "appleorangeYESNO", + "", + ), + PromptPair("YES", ""), + ) + + def test_cmd_if_var_vs_var_contains(self): # var contains var: var1 contains var2's value + self.process( + PromptPair( + "hello worldhelloYESNO", + "", + ), + PromptPair("YES", ""), + ) + + def test_cmd_if_var_vs_var_not_contains(self): # var not contains var: var1 does not contain var2's value + self.process( + PromptPair( + "hello worldgoodbyeYESNO", + "", + ), + PromptPair("YES", ""), + ) if __name__ == "__main__": unittest.main()