aljaz/tiingo-python
1
1
Fork 0
mirror of https://github.com/hydrosquall/tiingo-python.git synced 2025-12-18 12:24:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update test to support both python 2 and python 3

Browse Source
This commit is contained in:
Cameron Yick
2020-12-12 20:10:24 -05:00
parent e6f3fe205f
commit ea891a1458
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_restclient.py
View File

@@ -34,7 +34,14 @@ class TestRestClient(TestCase):
# Test 404 error
def test_invalid_url(self):
with self.assertRaisesRegex(RestClientError, "404"),\
# Suppress deprecation warning in python 3
if hasattr(self, 'assertRaisesRegex'):
assertFunc = self.assertRaisesRegex
else:
assertFunc = self.assertRaisesRegexp
with assertFunc(RestClientError, "404"),\
vcr.use_cassette('tests/fixtures/invalid_url.yaml'):
# Should return 404 error
self._client._request('GET', "bing_is_great")
@@ -66,7 +73,14 @@ class TestRestClientWithSession(TestCase):
# Test 404 error
def test_invalid_url(self):
with self.assertRaisesRegex(RestClientError, "404"),\
# Suppress deprecation warning in python 3
if hasattr(self, 'assertRaisesRegex'):
assertFunc = self.assertRaisesRegex
else:
assertFunc = self.assertRaisesRegexp
with assertFunc(RestClientError, "404"),\
vcr.use_cassette('tests/fixtures/invalid_url.yaml'):
# Should return 404 error
self._client._request('GET', "bing_is_great")