Skip to content

Commit e583bcb

Browse files
committed
[testDataView] add test for 2D string data
1 parent b3c21c3 commit e583bcb

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

nixio/test/test_data_view.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,21 @@ def test_data_view_read(self):
3333
dv = da.get_slice((10, 3), extents=(1, 3))
3434
np.testing.assert_almost_equal(dv[:], da[10:11, 3:6])
3535

36+
def test_data_view_read_2d_str_array(self):
37+
numbers = ["one", "two", "three", "four", "five", "six", "seven", "eight", "nine", "ten"]
38+
data = np.empty((10, 4), dtype=str)
39+
for i in range(10):
40+
for j in range(4):
41+
data[i, j] = " ".join([numbers[i], numbers[j]])
42+
da = self.file.blocks[0].create_data_array("str_2d_array", "test", dtype=nix.DataType.String, data=data)
43+
da.append_set_dimension()
44+
da.append_set_dimension()
45+
46+
dv = da.get_slice((0, 0), extents=(10, 4))
47+
npeq = np.testing.assert_equal
48+
npeq(dv.shape, da.shape)
49+
npeq(dv[:], data)
50+
3651
def test_data_view_fancy_slicing(self):
3752
da = self.file.blocks[0].data_arrays[0]
3853
dv = da.get_slice((5, 8), extents=(10, 20))

0 commit comments

Comments
 (0)