tests/test_inp.py

   1 #! /usr/bin/env python
   2
   3
   4 from tests import TestCase, main
   5 from m_librarian.db import Author, Book
   6
   7
   8 class TestInp(TestCase):
   9     def test_import_bad_inpx(self):
  10         self.assertRaises(ValueError, self.import_inpx, 'bad.inpx')
  11
  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)
  16
  17
  18 if __name__ == "__main__":
  19     main()