summaryrefslogtreecommitdiff
path: root/tests/test_core.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_core.py')
-rw-r--r--tests/test_core.py7
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):