From adec33cfea063cb6de603feb30fc147a8f567a87 Mon Sep 17 00:00:00 2001 From: Yuval Adam <_@yuv.al> Date: Tue, 20 Feb 2018 15:02:51 +0100 Subject: Add another sample --- tests/test_core.py | 7 ++++--- 1 file 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): -- cgit v1.3.1