created Task class

json_tests.py

@@ -3,7 +3,7 @@ import unittest
 
 import numpy as np
 
-from main import load_task, grid2FOL, FOL2grid, FOL2prolog, prolog2FOL_array, run_prolog_program, array_and_plot_grid
+from main import load_task, FOL2grid, FOL2prolog, prolog2FOL_array
 from task import Task
 
 

main.py

@@ -7,6 +7,14 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import subprocess
 
+tasks = {
+    "lines": "0b148d64.json",
+    "grids": "90f3ed37.json",
+    "pour": "d4f3cd78.json",
+    "cross": "e21d9049.json",
+    "stripes": "f8c80d96.json"
+}
+
 
 def run_prolog_program(program, curr_dir=""):
     # Construct the command to run SICStus Prolog