@@ -18,9 +18,9 @@ def test_get_invalid_item(self):
1818
1919 def test_register_class (self ):
2020 cr = ClassRegistry ()
21- cr .register_class ('models.Company' )
22- from models import Company
23- self .assertEqual (cr ['models.Company' ], Company )
21+ cr .register_class ('tests. models.Company' )
22+ from tests . models import Company
23+ self .assertIs (cr ['tests. models.Company' ], Company )
2424
2525
2626class TestSeeder (unittest .TestCase ):
@@ -68,26 +68,26 @@ def test_seed_no_relationship(self):
6868 def test_seed_multiple_entities (self ):
6969 instance = [
7070 {
71- "model" : "models.Company" ,
71+ "model" : "tests. models.Company" ,
7272 "data" : {
7373 "name" : "Mike Corporation" ,
7474 "!employees" : {
75- "model" : "models.Employee" ,
75+ "model" : "tests. models.Employee" ,
7676 "data" : {
7777 }
7878 }
7979 }
8080 },
8181 {
82- "model" : "models.Company" ,
82+ "model" : "tests. models.Company" ,
8383 "data" : [
8484 {
8585
8686 }
8787 ]
8888 },
8989 {
90- "model" : "models.Company" ,
90+ "model" : "tests. models.Company" ,
9191 "data" : {
9292
9393 }
0 commit comments