converted test_viewsets to pytest style

This commit is contained in:
Asif Saifuddin Auvi 2016-11-23 18:27:51 +06:00
parent 3e1b31bb9b
commit 4ec8bc5de0

View File

@ -21,15 +21,15 @@ class InitializeViewSetsTestCase(TestCase):
})
response = my_view(request)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data, {'ACTION': 'LIST'})
assert response.status_code == status.HTTP_200_OK
assert response.data == {'ACTION': 'LIST'}
def test_initialize_view_set_with_empty_actions(self):
try:
BasicViewSet.as_view()
except TypeError as e:
self.assertEqual(str(e), "The `actions` argument must be provided "
"when calling `.as_view()` on a ViewSet. "
"For example `.as_view({'get': 'list'})`")
assert str(e) == "The `actions` argument must be provided "
"when calling `.as_view()` on a ViewSet. "
"For example `.as_view({'get': 'list'})`"
else:
self.fail("actions must not be empty.")