encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-06 13:30:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Convert asserts to pytest style

Browse Source
This commit is contained in:
Artem Muterko 2017-01-08 14:20:00 +02:00
parent dfcd557c54
commit 5e676f1bfa
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/test_renderers.py
View File

@ -6,6 +6,7 @@ import re
from collections import MutableMapping, OrderedDict from collections import MutableMapping, OrderedDict
from types import SimpleNamespace from types import SimpleNamespace
import pytest
from django.conf.urls import include, url from django.conf.urls import include, url
from django.core.cache import cache from django.core.cache import cache
from django.db import models from django.db import models
@ -281,7 +282,7 @@ class BaseRendererTests(TestCase):
""" """
BaseRenderer.render should raise NotImplementedError BaseRenderer.render should raise NotImplementedError
""" """
with self.assertRaises(NotImplementedError): with pytest.raises(NotImplementedError):
BaseRenderer().render('test') BaseRenderer().render('test')
@ -596,7 +597,7 @@ class StaticHTMLRendererTests(TestCase):
def test_static_renderer(self): def test_static_renderer(self):
data = '<html><body>text</body></html>' data = '<html><body>text</body></html>'
result = self.renderer.render(data) result = self.renderer.render(data)
self.assertEqual(data, result) assert result == data
def test_static_renderer_with_exception(self): def test_static_renderer_with_exception(self):
context = { context = {
@ -604,7 +605,7 @@ class StaticHTMLRendererTests(TestCase):
'request': SimpleNamespace() 'request': SimpleNamespace()
} }
result = self.renderer.render({}, renderer_context=context) result = self.renderer.render({}, renderer_context=context)
self.assertEqual(result, '500 Internal Server Error') assert result == '500 Internal Server Error'
class BrowsableAPIRendererTests(TestCase): class BrowsableAPIRendererTests(TestCase):
@ -613,14 +614,14 @@ class BrowsableAPIRendererTests(TestCase):
self.renderer = BrowsableAPIRenderer() self.renderer = BrowsableAPIRenderer()
def test_get_description_returns_empty_string_for_401_and_403_statuses(self): def test_get_description_returns_empty_string_for_401_and_403_statuses(self):
self.assertEqual('', self.renderer.get_description({}, status_code=401)) assert self.renderer.get_description({}, status_code=401) == ''
self.assertEqual('', self.renderer.get_description({}, status_code=403)) assert self.renderer.get_description({}, status_code=403) == ''
def test_get_filter_form_returns_none_if_data_is_not_list_instance(self): def test_get_filter_form_returns_none_if_data_is_not_list_instance(self):
dummy_view = SimpleNamespace(get_queryset=None, filter_backends=None) dummy_view = SimpleNamespace(get_queryset=None, filter_backends=None)
result = self.renderer.get_filter_form(data='not list', result = self.renderer.get_filter_form(data='not list',
view=dummy_view, request={}) view=dummy_view, request={})
self.assertIsNone(result) assert result is None
class AdminRendererTests(TestCase): class AdminRendererTests(TestCase):
@ -642,6 +643,6 @@ class AdminRendererTests(TestCase):
result = self.renderer.render(data={'test': 'test'}, result = self.renderer.render(data={'test': 'test'},
renderer_context=context) renderer_context=context)
self.assertEqual(result, '') assert result == ''
self.assertEqual(response.status_code, status.HTTP_303_SEE_OTHER) assert response.status_code == status.HTTP_303_SEE_OTHER
self.assertEqual(response['Location'], 'http://example.com') assert response['Location'] == 'http://example.com'