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)