Convert test_models to pytest

Author: shepherdjay

lists/tests/test_models.py

@@ -1,31 +1,33 @@
-from django.test import TestCase
 from lists.models import List, Item
 from django.core.exceptions import ValidationError
+import pytest
 
+pytestmark = pytest.mark.django_db
 
-class ItemModelTest(TestCase):
+
+class TestItemModel:
     def test_default_text(self):
         item = Item()
-        self.assertEqual(item.text, '')
+        assert '' == item.text
 
     def test_item_is_related_to_list(self):
         list_ = List.objects.create()
         item = Item()
         item.list = list_
         item.save()
-        self.assertIn(item, list_.item_set.all())
+        assert item in list_.item_set.all()
 
     def test_cannot_save_empty_list_items(self):
         list_ = List.objects.create()
         item = Item(list=list_, text='')
-        with self.assertRaises(ValidationError):
+        with pytest.raises(ValidationError):
             item.save()
             item.full_clean()
 
     def test_duplicate_items_are_invalid(self):
         list_ = List.objects.create()
         Item.objects.create(list=list_, text='bla')
-        with self.assertRaises(ValidationError):
+        with pytest.raises(ValidationError):
             item = Item(list=list_, text='bla')
             item.full_clean()
 
@@ -41,18 +43,15 @@ def test_list_ordering(self):
         item1 = Item.objects.create(list=list1, text='i1')
         item2 = Item.objects.create(list=list1, text='item 2')
         item3 = Item.objects.create(list=list1, text='3')
-        self.assertEqual(
-            list(Item.objects.all()),
-            [item1, item2, item3]
-        )
+        assert list(Item.objects.all()) == [item1, item2, item3]
 
     def test_string_representation(self):
         item = Item(text='some text')
-        self.assertEqual(str(item), 'some text')
+        assert 'some text' == str(item)
 
 
-class ListModelTest(TestCase):
+class TestListModel:
 
     def test_get_absolute_url(self):
         list_ = List.objects.create()
-        self.assertEqual(list_.get_absolute_url(), '/lists/%d/' % list_.id)
+        assert '/lists/%d/' % list_.id == list_.get_absolute_url()