Docs: Update TODO

[m_librarian.git] / tests / test_glst.py

diff --git a/tests/test_glst.py b/tests/test_glst.py
old mode 100755 (executable)
new mode 100644 (file)
index 5349bb3..15fafa0
--- a/tests/test_glst.py
+++ b/tests/test_glst.py
@@ -1,16 +1,9 @@
-#! /usr/bin/env python
 
-
-from tests import TestCase, main
+from dbutils import setup_module, teardown_module  # noqa
 from m_librarian.db import Genre
 from m_librarian.glst import import_glst
 
 
-class TestGlst(TestCase):
-    def test_import_glst(self):
-        import_glst()
-        self.assertEqual(Genre.select().count(), 340)
-
-
-if __name__ == "__main__":
-    main()
+def test_import_glst():
+    import_glst()
+    assert Genre.select().count() == 370