encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-04-29 05:23:43 +03:00
Code Issues Packages Projects Releases Wiki Activity

conerted multi-table-inheritence test asserts to pytest

Browse Source
This commit is contained in:
Asif Saifuddin Auvi 2017-01-04 17:46:55 +06:00
parent 7874bcabe9
commit 8fa5846e51
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
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,5 @@ class InheritedModelSerializationTests(TestCase):
'name2': 'child name',
}
serializer = DerivedModelSerializer(data=data)
self.assertEqual(serializer.is_valid(), True)
assert serializer.is_valid() is True