#!/usr/bin/env python3

# standard libraries

import json

import logging

import os

import subprocess

from pprint import pformat

def rgb_matches(text1, text2):

data1 = json.loads(text1)

data2 = json.loads(text2)

for (square_index, rgb1) in sorted(data1.items()):

try:

rgb2 = data2[square_index]

except KeyError:

return False

(r1, g1, b1) = rgb1

(r2, g2, b2) = rgb2

if abs(r1 - r2) >= 15 or abs(g1 - g2) >= 15 or abs(b1 - b2) >= 15:

return False

return True

def get_rgb_delta(text1, text2):

result = []

data1 = json.loads(text1)

data2 = json.loads(text2)

for (square_index, rgb1) in sorted(data1.items()):

try:

rgb2 = data2[square_index]

except KeyError:

result.append("%2d: %14s != ERROR" % (int(square_index), pformat(rgb1)))

continue

(r1, g1, b1) = rgb1

(r2, g2, b2) = rgb2

if abs(r1 - r2) >= 15 or abs(g1 - g2) >= 15 or abs(b1 - b2) >= 15:

result.append("%2d: %14s != %-14s" % (int(square_index), pformat(rgb1), pformat(rgb2)))

return "\n".join(result)

if __name__ == "__main__":

logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)5s: %(message)s")

logger = logging.getLogger(__name__)

# Color the errors and warnings in red

logging.addLevelName(logging.ERROR, "\033[91m %s\033[0m" % logging.getLevelName(logging.ERROR))

logging.addLevelName(logging.WARNING, "\033[91m%s\033[0m" % logging.getLevelName(logging.WARNING))

# To add a test case:

# - place the cube in the robot, solve it

# - in the log output grab the "RGB json" and save that in a file in test-data

# - in the log output grab the "Final cube for kociema", this is what you put in the entry in the test_cases tuple

test_cases = (

("2x2x2 random 01", "test-data/2x2x2-random-01.txt"),

("2x2x2 random 02", "test-data/2x2x2-random-02.txt"),

("2x2x2 random 03", "test-data/2x2x2-random-03.txt"),

("2x2x2 random 04", "test-data/2x2x2-random-04.txt"),

("3x3x3 random 01", "test-data/3x3x3-random-01.txt"),

("4x4x4 random 01", "test-data/4x4x4-random-01.txt"),

("4x4x4 random 02", "test-data/4x4x4-random-02.txt"),

("5x5x5 random 01", "test-data/5x5x5-random-01.txt"),

("5x5x5 random 02", "test-data/5x5x5-random-02.txt"),

("5x5x5 random 03", "test-data/5x5x5-random-03.txt"),

("5x5x5 random 04", "test-data/5x5x5-random-04.txt"),

("6x6x6 random 01", "test-data/6x6x6-random-01.txt"),

("6x6x6 random 02", "test-data/6x6x6-random-02.txt"),

("7x7x7 random 01", "test-data/7x7x7-random-01.txt"),

)

results = []

for (desc, filename) in test_cases:

test_dir = os.path.abspath(filename[0:-4]) + "/"

logger.info("filename: %s" % filename)

logger.info("test_dir: %s" % test_dir)

try:

output = (

subprocess.check_output(["./usr/bin/rubiks-cube-tracker.py", "--directory", test_dir])

.decode("ascii")

.splitlines()[0]

.strip()

)

except subprocess.CalledProcessError as e:

print(("ERROR: rubiks-cube-locator.py barfed on %s" % filename))

print(e)

results.append("\033[91mFAIL\033[0m: %s (Exception)" % desc)

continue

with open(filename, "r") as fh:

expected_output = fh.readlines()[0].strip()

if rgb_matches(expected_output, output):

results.append("\033[92mPASS\033[0m: %s" % desc)

else:

results.append("\033[91mFAIL\033[0m: %s" % desc)

results.append(get_rgb_delta(expected_output, output))

results.append(output)

# break

print(("\n".join(results)))