convert tests asserts to pytest style (encode#4696)

auvipy · tomchristie · commit 4b59ec27fa72 · 2016-11-23T13:17:00.000Z
diff --git a/tests/test_views.py b/tests/test_views.py
@@ -71,8 +71,8 @@ def test_400_parse_error(self):
         expected = {
             'detail': JSON_ERROR
         }
-        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
-        self.assertEqual(sanitise_json_error(response.data), expected)
+        assert response.status_code == status.HTTP_400_BAD_REQUEST
+        assert sanitise_json_error(response.data) == expected
 
 
 class FunctionBasedViewIntegrationTests(TestCase):
@@ -85,8 +85,8 @@ def test_400_parse_error(self):
         expected = {
             'detail': JSON_ERROR
         }
-        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
-        self.assertEqual(sanitise_json_error(response.data), expected)
+        assert response.status_code == status.HTTP_400_BAD_REQUEST
+        assert sanitise_json_error(response.data) == expected
 
 
 class TestCustomExceptionHandler(TestCase):
@@ -107,14 +107,14 @@ def test_class_based_view_exception_handler(self):
         request = factory.get('/', content_type='application/json')
         response = view(request)
         expected = 'Error!'
-        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
-        self.assertEqual(response.data, expected)
+        assert response.status_code == status.HTTP_400_BAD_REQUEST
+        assert response.data == expected
 
     def test_function_based_view_exception_handler(self):
         view = error_view
 
         request = factory.get('/', content_type='application/json')
         response = view(request)
         expected = 'Error!'
-        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
-        self.assertEqual(response.data, expected)
+        assert response.status_code == status.HTTP_400_BAD_REQUEST
+        assert response.data == expected
diff --git a/tests/test_viewsets.py b/tests/test_viewsets.py
@@ -21,15 +21,15 @@ def test_initialize_view_set_with_actions(self):
         })
 
         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.")
