Add test

tests/test_party_company.py

@@ -5,12 +5,40 @@ import unittest
 import trytond.tests.test_tryton
 from trytond.tests.test_tryton import ModuleTestCase, with_transaction
 from trytond.pool import Pool
+from trytond.modules.company.tests import create_company, set_company
 
 
 class PartyCompanyTestCase(ModuleTestCase):
     'Test Party Company module'
     module = 'party_company'
 
+    @with_transaction()
+    def test_party(self):
+        'Create party'
+        pool = Pool()
+        Party = pool.get('party.party')
+
+        party1, = Party.create([{
+                    'name': 'Party 1',
+                    }])
+        self.assert_(party1.id)
+        self.assertEqual(party1.company, None)
+
+    @with_transaction()
+    def test_party_company(self):
+        'Create party company'
+        pool = Pool()
+        Party = pool.get('party.party')
+
+        company = create_company()
+        with set_company(company):
+            party2, = Party.create([{
+                        'name': 'Party 2',
+                        }])
+            self.assert_(party2.id)
+            self.assertNotEqual(party2.company, None)
+
+
 def suite():
     suite = trytond.tests.test_tryton.suite()
     suite.addTests(unittest.TestLoader().loadTestsFromTestCase(