converted write_only_fields tests to pytest style

This commit is contained in:
Asif Saifuddin Auvi 2016-11-23 19:10:12 +06:00
parent 3e1b31bb9b
commit 3bc4138fe7

View File

@ -20,8 +20,8 @@ class WriteOnlyFieldTests(TestCase):
'password': '123' 'password': '123'
} }
serializer = self.Serializer(data=data) serializer = self.Serializer(data=data)
self.assertTrue(serializer.is_valid()) assert serializer.is_valid() is True
self.assertEqual(serializer.validated_data, data) assert serializer.validated_data == data
def write_only_fields_are_not_present_on_output(self): def write_only_fields_are_not_present_on_output(self):
instance = { instance = {
@ -29,4 +29,4 @@ class WriteOnlyFieldTests(TestCase):
'password': '123' 'password': '123'
} }
serializer = self.Serializer(instance) serializer = self.Serializer(instance)
self.assertEqual(serializer.data, {'email': 'foo@example.com'}) assert serializer.data == {'email': 'foo@example.com'}