def test_parse_array_assignment_zeros(self):
line = 'Map = zeros(3,3,2);'
arr = _parse_array_assignment(line)
arr_should_be = np.zeros((3,3,2))
np.testing.assert_array_equal(arr, arr_should_be)
def test_parse_array_assignment_2d(self):
line = 'Map = [1.0 2.0 ;3.0 4.0 ];'
arr = _parse_array_assignment(line)
arr_should_be = np.array([[1.0, 2.0], [3.0, 4.0]])
np.testing.assert_array_almost_equal(arr, arr_should_be)
