def test_convert_data_type_civis():
response = _create_mock_response({'foo': 'bar'}, {'header': 'val'})
data = convert_response_data_type(response, return_type='snake')
assert isinstance(data, Response)
assert data['foo'] == 'bar'
assert data.headers == {'header': 'val'}
def test_response_to_no_content_snake():
for code in [204, 205]:
raw_response = _create_empty_response(code, {'header1': 'val1'})
data = convert_response_data_type(raw_response, return_type='snake')
assert isinstance(data, Response)
assert data.json_data is None
assert data.headers == {'header1': 'val1'}
def test_response_to_no_content_snake():
# Test empty response handling for codes where we're likely to see them.
for code in [202, 204, 205]:
raw_response = _create_empty_response(code, {'header1': 'val1'})
data = convert_response_data_type(raw_response, return_type='snake')
assert isinstance(data, Response)
assert data.json_data is None
assert data.headers == {'header1': 'val1'}
def _call_api(self, method, path=None, params=None, data=None, **kwargs):
iterator = kwargs.pop('iterator', False)
if iterator:
return PaginatedResponse(path, params, self)
else:
resp = self._make_request(method, path, params, data, **kwargs)
resp = convert_response_data_type(resp,
return_type=self._return_type)
return resp
def test_convert_data_type_civis_list():
response = _create_mock_response([{'foo': 'bar'}, {'fizz': 'buzz'}],
{'header': 'val'})
data = convert_response_data_type(response, return_type='snake')
assert isinstance(data, list)
assert len(data) == 2
assert isinstance(data[0], Response)
assert data[0]['foo'] == 'bar'
assert data[0].headers == {'header': 'val'}
def test_convert_data_type_pandas_df():
response = _create_mock_response([{'foo': 'bar'}], None)
data = convert_response_data_type(response, return_type='pandas')
assert isinstance(data, pd.DataFrame)
assert data.equals(pd.DataFrame.from_records([{'foo': 'bar'}]))
def test_convert_data_type_pandas_series():
response = _create_mock_response({'foo': 'bar'}, None)
data = convert_response_data_type(response, return_type='pandas')
assert isinstance(data, pd.Series)
assert data.equals(pd.Series({'foo': 'bar'}))
def test_convert_data_type_raw_parsed():
response = {'foo': 'bar'}
data = convert_response_data_type(response, return_type='raw')
assert isinstance(data, dict)
assert data == {'foo': 'bar'}
def test_convert_data_type_raw_unparsed():
response = _create_mock_response({}, {'header1': 'val1'})
data = convert_response_data_type(response, return_type='raw')
assert isinstance(data, requests.Response)
assert data.headers == {'header1': 'val1'}
