def test_make_indices_matrix_2_dims(self):
expected = np.vstack((np.tile(np.arange(2), 3),
np.repeat(np.arange(3), 2)))
ret_val = write_utils.make_indices_matrix([2, 3], is_position=False)
self.assertTrue(np.allclose(expected, ret_val))
ret_val = write_utils.make_indices_matrix([2, 3], is_position=True)
self.assertTrue(np.allclose(expected.T, ret_val))
def test_make_indices_matrix_1_dims(self):
expected = np.expand_dims(np.arange(4), axis=0)
ret_val = write_utils.make_indices_matrix([4], is_position=False)
self.assertTrue(np.allclose(expected, ret_val))
ret_val = write_utils.make_indices_matrix([4], is_position=True)
self.assertTrue(np.allclose(expected.T, ret_val))
def test_make_indices_weird_inputs(self):
with self.assertRaises(ValueError):
_ = write_utils.make_indices_matrix([2, 'hello', 3])
def test_make_indices_not_list(self):
with self.assertRaises(TypeError):
_ = write_utils.make_indices_matrix(1)
def test_make_indices_non_int_dim_sizes(self):
with self.assertRaises(ValueError):
_ = write_utils.make_indices_matrix([1.233, 2.4, 3])
def test_make_indices_dim_w_val_1(self):
with self.assertRaises(ValueError):
_ = write_utils.make_indices_matrix([1, 2, 3])