def test_get_color_on_face():
cube = Cube()
for face in FACE_ORDER:
assert cube.get_color_on_face(face) == face
cube.rotate(CubicRotation.ry)
assert cube.get_color_on_face("F") == "L"
assert cube.get_color_on_face("B") == "R"
assert cube.get_color_on_face("U") == "U"
assert cube.get_color_on_face("D") == "D"
assert cube.get_color_on_face("L") == "B"
assert cube.get_color_on_face("R") == "F"
cube.rotate(CubicRotation.rx)
assert cube.get_color_on_face("F") == "U"
assert cube.get_color_on_face("R") == "F"
assert cube.get_color_on_face("B") == "D"
def test_rotate():
cube = Cube()
rot = CubicRotation.ry
cube.rotate(rot)
assert np.array_equal(cube.rotation.apply(X), -Z)
assert np.array_equal(cube.rotation.apply(Y), Y)