encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-22 17:47:04 +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')
serializer = DerivedModelSerializer(child)
self.assertEqual(set(serializer.data.keys()),
set(['name1', 'name2', 'id']))
assert set(serializer.data.keys()) == set(['name1', 'name2', 'id'])
def test_onetoone_primary_key_model_fields_as_expected(self):
"""
@ -55,8 +54,7 @@ class InheritedModelSerializationTests(TestCase):
parent = ParentModel.objects.create(name1='parent name')
associate = AssociatedModel.objects.create(name='hello', ref=parent)
serializer = AssociatedModelSerializer(associate)
self.assertEqual(set(serializer.data.keys()),
set(['name', 'ref']))
assert set(serializer.data.keys()) == set(['name', 'ref'])
def test_data_is_valid_without_parent_ptr(self):
"""
@ -68,4 +66,4 @@ class InheritedModelSerializationTests(TestCase):
'name2': 'child name',
}
serializer = DerivedModelSerializer(data=data)
self.assertEqual(serializer.is_valid(), True)
assert serializer.is_valid() is True