diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_parse.py | 2 | ||||
| -rw-r--r-- | tests/test_viewstate.py | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_parse.py b/tests/test_parse.py index c1e1abb..74e6675 100644 --- a/tests/test_parse.py +++ b/tests/test_parse.py @@ -3,7 +3,7 @@ import pytest from datetime import datetime from os import walk from os.path import join -from viewstate import * +from viewstate import ViewState, ViewStateException class TestParse(object): diff --git a/tests/test_viewstate.py b/tests/test_viewstate.py index 628e1e0..db01e14 100644 --- a/tests/test_viewstate.py +++ b/tests/test_viewstate.py @@ -2,7 +2,7 @@ import pytest from os import walk from os.path import join -from viewstate import * +from viewstate import ViewState, ViewStateException class TestViewState(object): @@ -19,7 +19,7 @@ class TestViewState(object): def test_invalid_base64(self): with pytest.raises(ViewStateException): - vs = ViewState("hello") + ViewState("hello") def test_invalid_decode(self): with pytest.raises(ViewStateException): |