def test_read_csv(tmpdir): expected_filename = os.path.join(tmpdir, 'test.csv') column_names = ['column_1', 'column_2', 'column_3'] expected_data = np.random.random((5, len(column_names))) # Write csv file io.write_csv(expected_filename, expected_data, column_names=column_names) assert os.path.exists(expected_filename) # Read csv file read_data, read_column_names, _ = io.read_csv(expected_filename) read_data = np.array(read_data).astype('float') np.testing.assert_allclose(expected_data, read_data) assert column_names == read_column_names
def test_write_csv(tmpdir): expected_filename = os.path.join(tmpdir, 'test.csv') column_names = ['column_1', 'column_2', 'column_3'] expected_data = np.random.random((5, len(column_names))) # Write csv file io.write_csv(expected_filename, expected_data, column_names=column_names) assert os.path.exists(expected_filename) # Check csv file is as expected with open(expected_filename) as output_csv: csv.reader(output_csv, delimiter=',') for row_index, row in enumerate(output_csv): if row_index == 0: assert row == "column_1,column_2,column_3\n" else: output_row_data = [float(i) for i in row.split(',')] np.testing.assert_allclose(np.array(output_row_data), expected_data[row_index - 1])
def test_builtin_reader_plugin_csv(viewer_factory, tmpdir): """Test the builtin reader plugin reads a temporary file.""" tmp = os.path.join(tmpdir, 'test.csv') column_names = ['index', 'axis-0', 'axis-1'] table = np.random.random((5, 3)) data = table[:, 1:] # Write csv file io.write_csv(tmp, table, column_names=column_names) layer_data = read_data_with_plugins(tmp) assert isinstance(layer_data, list) assert len(layer_data) == 1 assert isinstance(layer_data[0], tuple) assert layer_data[0][2] == 'points' assert np.allclose(data, layer_data[0][0]) view, viewer = viewer_factory() viewer.open(tmp, plugin='builtins') assert np.allclose(viewer.layers[0].data, data)