def test_generate_trajectory(self):
# Two groups of three factors
G = np.array([[1, 0], [0, 1], [0, 1]])
# Four levels
num_levels = 4
output = generate_trajectory(G, num_levels)
if np.any((output > 1) | (output < 0)):
raise AssertionError("Bound not working: %s", output)
assert_equal(output.shape[0], 3)
assert_equal(output.shape[1], 3)
def test_generate_trajectory(self):
# Two groups of three factors
G = np.array([[1, 0], [0, 1], [0, 1]])
# Four levels
num_levels = 4
output = generate_trajectory(G, num_levels)
if np.any((output > 1) | (output < 0)):
raise AssertionError("Bound not working: %s", output)
assert_equal(output.shape[0], 3)
assert_equal(output.shape[1], 3)