def test_default_output_fn_gpu(): tensor_gpu = torch.LongTensor([[1, 2, 3], [4, 5, 6]]).cuda() output = default_output_fn(tensor_gpu, content_types.CSV) assert '1,2,3\n4,5,6\n' in output.get_data(as_text=True) assert content_types.CSV in output.content_type
def test_default_output_fn_npy(tensor): output = default_output_fn(tensor, content_types.NPY) stream = BytesIO() np.save(stream, tensor.cpu().numpy()) assert stream.getvalue() in output.get_data(as_text=False) assert content_types.NPY in output.content_type
def test_default_output_fn_bad_accept(): with pytest.raises(encoders.UnsupportedFormatError): default_output_fn('', 'application/not_supported')
def test_default_output_fn_csv_float(): tensor = torch.FloatTensor([[1, 2, 3], [4, 5, 6]]) output = default_output_fn(tensor, content_types.CSV) assert '1.0,2.0,3.0\n4.0,5.0,6.0\n' in output.get_data(as_text=True) assert content_types.CSV in output.content_type
def test_default_output_fn_json(tensor): output = default_output_fn(tensor, content_types.JSON) assert json.dumps(tensor.cpu().numpy().tolist()) in output.get_data(as_text=True) assert content_types.JSON in output.content_type
def test_default_output_fn_csv_long(): tensor = torch.LongTensor([[1, 2, 3], [4, 5, 6]]) output = default_output_fn(tensor, content_types.CSV) assert '1,2,3\n4,5,6\n' in output.get_data(as_text=True) assert content_types.CSV == output.mimetype