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'}