Migrate to pytest

3 files changed, 30 insertions, 28 deletions

diff --git a/.gitignore b/.gitignore
index bee8a64..b943e2c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,3 @@
 __pycache__
+*.pyc
+.pytest_cache
diff --git a/tests/__init__.py b/tests/__init__.py
index f203615..e69de29 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -1,28 +0,0 @@
-import unittest
-
-from viewstate import ViewState
-
-class TestViewState(unittest.TestCase):
-
-    def test_blank(self):
-        vs = ViewState()
-        self.assertFalse(vs.is_valid())
-
-    def test_is_valid(self):
-        with open('samples/ngcs.sample', 'r') as f:
-            vs = ViewState(f.read())
-            self.assertTrue(vs.is_valid())
-
-    def test_invalid_decode(self):
-        with self.assertRaises(Exception):
-            vs = ViewState()
-            vs.raw = b'\x01\x02'
-            vs.decode()
-
-    def test_parse_const_value(self):
-        vs = ViewState()
-        vs.raw = b'\xff\x01\x67'
-        self.assertEqual(vs.decode(), True)
-
-if __name__ == '__main__':
-    unittest.main()
diff --git a/tests/test_core.py b/tests/test_core.py
new file mode 100644
index 0000000..3eed24c
--- /dev/null
+++ b/tests/test_core.py
@@ -0,0 +1,28 @@
+import pytest
+
+from viewstate import ViewState
+
+
+class TestViewState(object):
+
+    def test_blank(self):
+        vs = ViewState()
+        assert not vs.is_valid()
+
+    @pytest.mark.skip(reason='Missing test data')
+    def test_is_valid(self):
+        with open('samples/ngcs.sample', 'r') as f:
+            vs = ViewState(f.read())
+            self.assertTrue(vs.is_valid())
+
+    def test_invalid_decode(self):
+        with pytest.raises(Exception):
+            vs = ViewState()
+            vs.raw = b'\x01\x02'
+            vs.decode()
+
+    def test_parse_const_value(self):
+        vs = ViewState()
+        vs.raw = b'\xff\x01\x67'
+        assert vs.decode() is True
+