Fix mock assertions in migration tests

Author: codingjoe

tests/test_migrations.py

@@ -68,7 +68,7 @@ class Meta:
         monkeypatch.setattr(migration, "to_picture_field", Mock())
         migration.alter_picture_field(FromModel, ToModel)
 
-        assert migration.to_picture_field.called_once_with(ToModel)
+        migration.to_picture_field.assert_called_once_with(FromModel, ToModel)
 
     def test_alter_picture_field__picture_to_image(self, request, monkeypatch):
         class FromModel(models.Model):
@@ -87,7 +87,7 @@ class Meta:
         monkeypatch.setattr(migration, "from_picture_field", Mock())
         migration.alter_picture_field(FromModel, ToModel)
 
-        assert migration.from_picture_field.called_once_with(FromModel)
+        migration.from_picture_field.assert_called_once_with(FromModel)
 
     def test_alter_picture_field__picture_to_picture(self, request, monkeypatch):
         class FromModel(models.Model):
@@ -108,7 +108,7 @@ class Meta:
         monkeypatch.setattr(migration, "to_picture_field", Mock())
         migration.alter_picture_field(FromModel, ToModel)
         from_field = FromModel._meta.get_field("picture")
-        assert migration.update_pictures.called_once_with(from_field, ToModel)
+        migration.update_pictures.assert_called_once_with(from_field, ToModel)
         assert not migration.from_picture_field.called
         assert not migration.to_picture_field.called