From fb23cef254e6ed89795bc49f3ed3821472df8365 Mon Sep 17 00:00:00 2001 From: Carlton Gibson Date: Thu, 14 Feb 2019 16:36:10 +0100 Subject: [PATCH] Skip malformed YAML on PY2 test. --- tests/test_generateschema.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_generateschema.py b/tests/test_generateschema.py index 047bb406a..915c6ea05 100644 --- a/tests/test_generateschema.py +++ b/tests/test_generateschema.py @@ -5,7 +5,7 @@ from django.conf.urls import url from django.core.management import call_command from django.test import TestCase from django.test.utils import override_settings -from six import StringIO +from django.utils import six from rest_framework.compat import coreapi from rest_framework.utils import formatting, json @@ -28,8 +28,9 @@ class GenerateSchemaTests(TestCase): """Tests for management command generateschema.""" def setUp(self): - self.out = StringIO() + self.out = six.StringIO() + @pytest.mark.skipif(six.PY2, reason='PyYAML unicode output is malformed on PY2.') def test_renders_default_schema_with_custom_title_url_and_description(self): expected_out = """info: description: Sample description