def test_dataframe_handle_request_csv():
def test_function(df):
return df["name"][0]
input_adapter = DataframeInput()
csv_data = 'name,game,city\njohn,mario,sf'
request = MagicMock(spec=flask.Request)
request.headers = (('orient', 'records'), )
request.content_type = 'text/csv'
request.get_data.return_value = csv_data
result = input_adapter.handle_request(request, test_function)
assert result.get_data().decode('utf-8') == '"john"'
def test_dataframe_handle_request_csv():
def test_function(df):
return df["name"][0]
input_adapter = DataframeInput()
csv_data = 'name,game,city\njohn,mario,sf'.encode('utf-8')
request = Mock()
request.headers = {'orient': 'records'}
request.content_type = 'text/csv'
request.data = csv_data
result = input_adapter.handle_request(request, test_function)
assert result.get_data().decode('utf-8') == '"john"'
