4 from tests import TestCase, main
5 from m_librarian.db import Author, Book
8 class TestInp(TestCase):
9 def test_import_bad_inpx(self):
10 self.assertRaises(ValueError, self.import_inpx, 'bad.inpx')
12 def test_import_inpx(self):
13 self.import_inpx('test.inpx')
14 self.assertEqual(Author.select().count(), 4)
15 self.assertEqual(Book.select().count(), 4)
18 if __name__ == "__main__":