converted asserts of atomic requests test to pytest

auvipy · auvipy · commit 504f4b44c6dc · 2016-11-30T16:17:30.000+06:00
diff --git a/tests/test_atomic_requests.py b/tests/test_atomic_requests.py
@@ -67,8 +67,8 @@ def test_no_exception_commit_transaction(self):
 
         with self.assertNumQueries(1):
             response = self.view(request)
-        self.assertFalse(transaction.get_rollback())
-        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        assert not transaction.get_rollback()
+        assert response.status_code == status.HTTP_200_OK
         assert BasicModel.objects.count() == 1
 
 
@@ -98,7 +98,7 @@ def test_generic_exception_delegate_transaction_management(self):
             # 3 - release savepoint
             with transaction.atomic():
                 self.assertRaises(Exception, self.view, request)
-                self.assertFalse(transaction.get_rollback())
+                assert not transaction.get_rollback()
         assert BasicModel.objects.count() == 1
 
 
@@ -128,9 +128,8 @@ def test_api_exception_rollback_transaction(self):
             # 4 - release savepoint (django>=1.8 only)
             with transaction.atomic():
                 response = self.view(request)
-                self.assertTrue(transaction.get_rollback())
-        self.assertEqual(response.status_code,
-                         status.HTTP_500_INTERNAL_SERVER_ERROR)
+                assert transaction.get_rollback()
+        assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
         assert BasicModel.objects.count() == 0
 
 
@@ -151,5 +150,4 @@ def test_api_exception_rollback_transaction_non_atomic_view(self):
 
         # without checking connection.in_atomic_block view raises 500
         # due attempt to rollback without transaction
-        self.assertEqual(response.status_code,
-                         status.HTTP_404_NOT_FOUND)
+        assert response.status_code == status.HTTP_404_NOT_FOUND
