Skip to content
Snippets Groups Projects
Commit 3045d2f2 authored by tbuckworth's avatar tbuckworth
Browse files

finally got tests working, but lution is wrong

parent 5d273ddf
Branches recruiter-internship-page
No related tags found
No related merge requests found
......@@ -3,7 +3,7 @@ import unittest
import numpy as np
from main import load_task, grid2FOL, FOL2grid, FOL2prolog, prolog2FOL_array, run_prolog_program
from main import load_task, grid2FOL, FOL2grid, FOL2prolog, prolog2FOL_array, run_prolog_program, array_and_plot_grid
class TaskTester(unittest.TestCase):
......@@ -46,9 +46,12 @@ class TaskTester(unittest.TestCase):
out_FOL_array = prolog2FOL_array(out_prolog)
out_grid = FOL2grid(out_FOL_array)
array_and_plot_grid(self.output_grid)
array_and_plot_grid(out_grid)
self.assertTrue((out_grid == self.output_grid).all())
print(out)
print("done")
......
......@@ -167,7 +167,7 @@ def array_and_plot_grid(input_grid):
def grid2rgb(input_grid):
grid = np.array(input_grid)
grid = np.array(input_grid).astype(np.int64)
rgb_grid = rgb_lookup()[grid]
return rgb_grid
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment