diff options
| author | Yuval Adam <_@yuv.al> | 2018-02-20 15:02:51 +0100 |
|---|---|---|
| committer | Yuval Adam <_@yuv.al> | 2018-02-20 15:02:51 +0100 |
| commit | adec33cfea063cb6de603feb30fc147a8f567a87 (patch) | |
| tree | 542867a5167c08b67507538e3995c10854864470 | |
| parent | 8638d1c254e7813de469e86eaa290b413c547ff5 (diff) | |
Add another sample
| -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): |
