-#! /usr/bin/env python
-
-import os
-from tests import TestCase, main
+from pytest import raises
+from dbutils import setup_module, teardown_module # noqa
+from dbutils import load_inpx
from m_librarian.db import Author, Book
-from m_librarian.inp import import_inpx
-
-class TestInp(TestCase):
- def test_import_bad_inpx(self):
- self.assertRaises(
- ValueError, import_inpx,
- os.path.join(os.path.dirname(__file__), 'bad.inpx'))
- def test_import_inpx(self):
- import_inpx(os.path.join(os.path.dirname(__file__), 'test.inpx'))
- self.assertEqual(Author.select().count(), 4)
- self.assertEqual(Book.select().count(), 4)
+def test_import_bad_inpx():
+ raises(ValueError, load_inpx, 'bad.inpx')
-if __name__ == "__main__":
- main()
+def test_import_inpx():
+ load_inpx('test.inpx')
+ assert Author.select().count() == 5
+ assert Book.select().count() == 5