encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-23 01:57:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4780 from auvipy/multi

Browse Source 
converted multi-table-inheritence test asserts to pytest
This commit is contained in:
Xavier Ordoquy 2017-01-04 13:16:14 +01:00 committed by GitHub
commit 0f4cd2710c
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_multitable_inheritance.py
View File

@ -44,8 +44,7 @@ class InheritedModelSerializationTests(TestCase):
""" """
child = ChildModel(name1='parent name', name2='child name') child = ChildModel(name1='parent name', name2='child name')
serializer = DerivedModelSerializer(child) serializer = DerivedModelSerializer(child)
self.assertEqual(set(serializer.data.keys()), assert set(serializer.data.keys()) == set(['name1', 'name2', 'id'])
set(['name1', 'name2', 'id']))
def test_onetoone_primary_key_model_fields_as_expected(self): def test_onetoone_primary_key_model_fields_as_expected(self):
""" """
@ -55,8 +54,7 @@ class InheritedModelSerializationTests(TestCase):
parent = ParentModel.objects.create(name1='parent name') parent = ParentModel.objects.create(name1='parent name')
associate = AssociatedModel.objects.create(name='hello', ref=parent) associate = AssociatedModel.objects.create(name='hello', ref=parent)
serializer = AssociatedModelSerializer(associate) serializer = AssociatedModelSerializer(associate)
self.assertEqual(set(serializer.data.keys()), assert set(serializer.data.keys()) == set(['name', 'ref'])
set(['name', 'ref']))
def test_data_is_valid_without_parent_ptr(self): def test_data_is_valid_without_parent_ptr(self):
""" """
@ -68,4 +66,4 @@ class InheritedModelSerializationTests(TestCase):
'name2': 'child name', 'name2': 'child name',
} }
serializer = DerivedModelSerializer(data=data) serializer = DerivedModelSerializer(data=data)
self.assertEqual(serializer.is_valid(), True) assert serializer.is_valid() is True