diff --git a/tests/unit/conditions/test_return_value.py b/tests/unit/conditions/test_return_value.py index cfbb2eec8..8303d66db 100644 --- a/tests/unit/conditions/test_return_value.py +++ b/tests/unit/conditions/test_return_value.py @@ -4,11 +4,21 @@ import pytest from nucypher.policy.conditions.lingo import ReturnValueTest -def test_return_result_test_simple(): +def test_return_value_test_integer(): test = ReturnValueTest(comparator='>', value='0') assert test.eval('1') assert not test.eval('-1') + test = ReturnValueTest(comparator='>', value=0) + assert test.eval(1) + assert not test.eval(-1) + + +def test_return_value_test_string(): + test = ReturnValueTest(comparator='==', value='"foo"') + assert test.eval('"foo"') + assert not test.eval('"bar"') + def test_return_value_sanitization(): with pytest.raises(ValueError):