Skip to content

Commit 3cb26cc

Browse files
committed
Update tests
1 parent eefc615 commit 3cb26cc

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/test_seeder.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

2626
class 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

Comments
 (0)