From f575f0e54d3440131e54ff34b05e2af0a3e98946 Mon Sep 17 00:00:00 2001 From: Daniel Grossmann-Kavanagh Date: Tue, 14 Jan 2020 16:08:35 -0800 Subject: [PATCH] update tests to allow new datetime behavior in PyYAML 5.3 --- tests/aiohttp/test_aiohttp_datetime.py | 5 ++++- tests/test_flask_encoder.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/aiohttp/test_aiohttp_datetime.py b/tests/aiohttp/test_aiohttp_datetime.py index 99a2c40d6..73d75b101 100644 --- a/tests/aiohttp/test_aiohttp_datetime.py +++ b/tests/aiohttp/test_aiohttp_datetime.py @@ -27,7 +27,10 @@ def get_value(data, path): return data example = get_value(spec_data, 'paths./datetime.get.responses.200.content.application/json.schema.example.value') - assert example == '2000-01-23T04:56:07.000008Z' + assert example in [ + '2000-01-23T04:56:07.000008+00:00', # PyYAML 5.3 + '2000-01-23T04:56:07.000008Z' + ] example = get_value(spec_data, 'paths./date.get.responses.200.content.application/json.schema.example.value') assert example == '2000-01-23' example = get_value(spec_data, 'paths./uuid.get.responses.200.content.application/json.schema.example.value') diff --git a/tests/test_flask_encoder.py b/tests/test_flask_encoder.py index e08f31838..39c4a7e85 100644 --- a/tests/test_flask_encoder.py +++ b/tests/test_flask_encoder.py @@ -63,7 +63,10 @@ def get_value(data, path): return data example = get_value(spec_data, 'paths./datetime.get.{}.example.value'.format(response_path)) - assert example == '2000-01-23T04:56:07.000008Z' + assert example in [ + '2000-01-23T04:56:07.000008+00:00', # PyYAML 5.3+ + '2000-01-23T04:56:07.000008Z' + ] example = get_value(spec_data, 'paths./date.get.{}.example.value'.format(response_path)) assert example == '2000-01-23' example = get_value(spec_data, 'paths./uuid.get.{}.example.value'.format(response_path))