diff options
Diffstat (limited to 'tests/test_core.py')
| -rw-r--r-- | tests/test_core.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/test_core.py b/tests/test_core.py index 2a0078a..0c71a62 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -10,9 +10,10 @@ class TestViewState(object): assert not vs.is_valid() def test_is_valid(self): - with open('tests/samples/ngcs.sample', 'r') as f: - vs = ViewState(f.read()) - assert vs.is_valid() is True + for s in ['ngcs', 'mot']: + with open(f'tests/samples/{s}.sample', 'r') as f: + vs = ViewState(f.read()) + assert vs.is_valid() is True def test_invalid_base64(self): with pytest.raises(ViewStateException): |
