Fix unittest for new tests module

tests/test_company.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
-#This file is part of Tryton.  The COPYRIGHT file at the top level of this repository contains the full copyright notices and license terms.
+#This file is part of Tryton.  The COPYRIGHT file at the top level of
+#this repository contains the full copyright notices and license terms.
 
 import sys, os
 DIR = os.path.abspath(os.path.normpath(os.path.join(__file__,
@@ -8,8 +9,8 @@ if os.path.isdir(DIR):
     sys.path.insert(0, os.path.dirname(DIR))
 
 import unittest
-import trytond.tests
-from trytond.tests import RPCProxy, CONTEXT, SOCK
+import trytond.tests.test_tryton
+from trytond.tests.test_tryton import RPCProxy, CONTEXT, SOCK
 
 
 class CompanyTestCase(unittest.TestCase):
@@ -18,7 +19,7 @@ class CompanyTestCase(unittest.TestCase):
     '''
 
     def setUp(self):
-        trytond.tests.install_module('company')
+        trytond.tests.test_tryton.install_module('company')
         self.company = RPCProxy('company.company')
         self.employee = RPCProxy('company.employee')
         self.currency = RPCProxy('currency.currency')
@@ -78,7 +79,7 @@ def suite():
     return unittest.TestLoader().loadTestsFromTestCase(CompanyTestCase)
 
 if __name__ == '__main__':
-    suiteTrytond = trytond.tests.suite()
+    suiteTrytond = trytond.tests.test_tryton.suite()
     suiteCompany = suite()
     alltests = unittest.TestSuite([suiteTrytond, suiteCompany])
     unittest.TextTestRunner(verbosity=2).run(alltests)