Tests

tests/test_country_zip.py

@@ -1,46 +1,18 @@
-#!/usr/bin/env python
-#This file is part helloword module for Tryton.
-#The COPYRIGHT file at the top level of this repository contains 
-#the full copyright notices and license terms.
-
-import sys
-import os
-DIR = os.path.abspath(os.path.normpath(os.path.join(__file__,
-    '..', '..', '..', '..', '..', 'trytond')))
-if os.path.isdir(DIR):
-    sys.path.insert(0, os.path.dirname(DIR))
-
+# This file is part of the country_zip module for Tryton.
+# The COPYRIGHT file at the top level of this repository contains the full
+# copyright notices and license terms.
 import unittest
 import trytond.tests.test_tryton
-from trytond.tests.test_tryton import test_view, test_depends
+from trytond.tests.test_tryton import ModuleTestCase
 
 
-class CountryZipTestCase(unittest.TestCase):
-    '''
-    Test Country Zip module.
-    '''
-
-    def setUp(self):
-        trytond.tests.test_tryton.install_module('country_zip')
-
-    def test0005views(self):
-        '''
-        Test views.
-        '''
-        test_view('country_zip')
-
-    def test0006depends(self):
-        '''
-        Test depends.
-        '''
-        test_depends()
+class CountryZipTestCase(ModuleTestCase):
+    'Test Country Zip module'
+    module = 'country_zip'
 
 
 def suite():
     suite = trytond.tests.test_tryton.suite()
     suite.addTests(unittest.TestLoader().loadTestsFromTestCase(
         CountryZipTestCase))
-    return suite
-
-if __name__ == '__main__':
-    unittest.TextTestRunner(verbosity=2).run(suite())
+    return suite