From 76ac641fbd6c9d7dff5da3c551c3fd1ef7dedd2e Mon Sep 17 00:00:00 2001
From: Tom Christie
Date: Tue, 2 Dec 2014 13:04:49 +0000
Subject: Minor tweaks for helpful message on Model.objects.create() failure.

---
 tests/test_model_serializer.py | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

(limited to 'tests')

diff --git a/tests/test_model_serializer.py b/tests/test_model_serializer.py
index 90767dac..1bcd58e0 100644
--- a/tests/test_model_serializer.py
+++ b/tests/test_model_serializer.py
@@ -10,7 +10,6 @@ from django.core.validators import MaxValueValidator, MinValueValidator, MinLeng
 from django.db import models
 from django.test import TestCase
 from rest_framework import serializers
-import pytest
 
 
 def dedent(blocktext):
@@ -87,13 +86,10 @@ class TestModelSerializer(TestCase):
             'non_model_field': 'bar',
         })
         serializer.is_valid()
-        with pytest.raises(TypeError):
+        with self.assertRaises(TypeError) as excinfo:
             serializer.save()
-
-        try:
-            serializer.save()
-        except TypeError as exc:
-            assert 'ModelSerializer' in str(exc)
+        msginitial = 'Got a `TypeError` when calling `OneFieldModel.objects.create()`.'
+        assert str(excinfo.exception).startswith(msginitial)
 
 
 class TestRegularFieldMappings(TestCase):
-- 
cgit v1.2.3