tests/test_inp.py

   1
   2 from pytest import raises
   3 from dbutils import setup_module, teardown_module  # noqa
   4 from dbutils import load_inpx
   5 from m_librarian.db import Author, Book
   6
   7
   8 def test_import_bad_inpx():
   9     raises(ValueError, load_inpx, 'bad.inpx')
  10
  11
  12 def test_import_inpx():
  13     load_inpx('test.inpx')
  14     assert Author.select().count() == 4
  15     assert Book.select().count() == 4