def test_get_data_vect(self):
r"""
Tests extraction of a vector from a data array
"""
data = sp.arange(100)
data = data.reshape(10, 10)
#
vect = amt.get_data_vect(data, 'x', 0)
assert sp.all(vect == data[0, :])
vect = amt.get_data_vect(data, 'x', 11)
assert sp.all(vect == data[9, :])
vect = amt.get_data_vect(data, 'z', 0)
assert sp.all(vect == data[:, 0])
vect = amt.get_data_vect(data, 'z', 11)
assert sp.all(vect == data[:, 9])
#
with pytest.raises(ValueError):
amt.get_data_vect(data, 'y')
