def test_inverse(self):
s = cube3x3.initial_state
for a in cube3x3.Action:
s = cube3x3.transform(s, a)
r = cube3x3.render(s)
s = cube3x3.transform(s, cube3x3.inverse_action(a))
r2 = cube3x3.render(s)
self.assertEqual(s, cube3x3.initial_state)
def test_inverse_back(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.B)
s = cube3x3.transform(s, cube3x3.Action.b)
self.assertEqual(s, cube3x3.initial_state)
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.b)
s = cube3x3.transform(s, cube3x3.Action.B)
self.assertEqual(s, cube3x3.initial_state)
def test_inverse_down(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.D)
s = cube3x3.transform(s, cube3x3.Action.d)
self.assertEqual(s, cube3x3.initial_state)
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.d)
s = cube3x3.transform(s, cube3x3.Action.D)
self.assertEqual(s, cube3x3.initial_state)
def test_inverse_front(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.F)
s = cube3x3.transform(s, cube3x3.Action.f)
self.assertEqual(s, cube3x3.initial_state)
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.f)
s = cube3x3.transform(s, cube3x3.Action.F)
self.assertEqual(s, cube3x3.initial_state)
def test_inverse_top(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.T)
s = cube3x3.transform(s, cube3x3.Action.t)
self.assertEqual(s, cube3x3.initial_state)
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.t)
s = cube3x3.transform(s, cube3x3.Action.T)
self.assertEqual(s, cube3x3.initial_state)
def test_inverse_left(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.L)
s = cube3x3.transform(s, cube3x3.Action.l)
self.assertEqual(s, cube3x3.initial_state)
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.l)
s = cube3x3.transform(s, cube3x3.Action.L)
self.assertEqual(s, cube3x3.initial_state)
def test_inverse_right(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.R)
s = cube3x3.transform(s, cube3x3.Action.r)
self.assertEqual(s, cube3x3.initial_state)
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.r)
s = cube3x3.transform(s, cube3x3.Action.R)
self.assertEqual(s, cube3x3.initial_state)
def test_sequence(self):
acts = [
cube3x3.Action.R, cube3x3.Action.t, cube3x3.Action.R,
cube3x3.Action.D, cube3x3.Action.F, cube3x3.Action.d,
cube3x3.Action.T, cube3x3.Action.R, cube3x3.Action.D,
cube3x3.Action.F
]
s = cube3x3.initial_state
for a in acts:
s = cube3x3.transform(s, a)
r = cube3x3.render(s)
for a in reversed(acts):
s = cube3x3.transform(s, cube3x3.inverse_action(a))
r = cube3x3.render(s)
self.assertEqual(s, cube3x3.initial_state)
def test_random(self):
s = cube3x3.initial_state
for _ in range(200):
a = cube3x3.Action(random.randrange(len(cube3x3.Action)))
s = cube3x3.transform(s, a)
tgt = np.zeros(shape=cube3x3.encoded_shape)
cube3x3.encode_inplace(tgt, s)
self.assertEqual(tgt.sum(), 20)
def test_left_rev(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.l)
r = cube3x3.render(s)
self.assertEqual(r.back, ['O', 'O', 'W'] * 3)
self.assertEqual(r.bottom, ['O', 'Y', 'Y'] * 3)
self.assertEqual(r.front, ['Y', 'R', 'R'] * 3)
self.assertEqual(r.left, ['G'] * 9)
self.assertEqual(r.right, ['B'] * 9)
self.assertEqual(r.top, ['R', 'W', 'W'] * 3)
def test_top(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.T)
r = cube3x3.render(s)
self.assertEqual(r.top, ['W'] * 9)
self.assertEqual(r.back, ['G'] * 3 + ['O'] * 6)
self.assertEqual(r.bottom, ['Y'] * 9)
self.assertEqual(r.front, ['B'] * 3 + ['R'] * 6)
self.assertEqual(r.left, ['R'] * 3 + ['G'] * 6)
self.assertEqual(r.right, ['O'] * 3 + ['B'] * 6)
def test_left_rev(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.l)
r = cube3x3.render(s)
self.assertEqual(r.back, ["O", "O", "W"] * 3)
self.assertEqual(r.bottom, ["O", "Y", "Y"] * 3)
self.assertEqual(r.front, ["Y", "R", "R"] * 3)
self.assertEqual(r.left, ["G"] * 9)
self.assertEqual(r.right, ["B"] * 9)
self.assertEqual(r.top, ["R", "W", "W"] * 3)
def test_top(self):
s = cube3x3.initial_state
s = cube3x3.transform(s, cube3x3.Action.T)
r = cube3x3.render(s)
self.assertEqual(r.top, ["W"] * 9)
self.assertEqual(r.back, ["G"] * 3 + ["O"] * 6)
self.assertEqual(r.bottom, ["Y"] * 9)
self.assertEqual(r.front, ["B"] * 3 + ["R"] * 6)
self.assertEqual(r.left, ["R"] * 3 + ["G"] * 6)
self.assertEqual(r.right, ["O"] * 3 + ["B"] * 6)
