add test

tests.py

@@ -1,17 +1,14 @@
 test_cases = [
     {
-        "predictions": [0, 0],
-        "references": [1, 1],
-        "result": {"metric_score": 0}
-    },
-    {
-        "predictions": [1, 1],
-        "references": [1, 1],
-        "result": {"metric_score": 1}
+        "candidates": ["def add(a,b):\n    return a*b\n", "def add(a, b):\n    return a+b\n"],
+        "references": ["if __name__ == \"__main__\":\n    print(add(2, 3))\n"],
+        "output" : ["5"]
+        "result": {"metric_score": 0.5}
     },
     {
-        "predictions": [1, 0],
-        "references": [1, 1],
-        "result": {"metric_score": 0.5}
+        "candidates": ["def add(a,b):\n    return a*b\n", "def add(a, b):\n    return a^b\n"],
+        "references": ["if __name__ == \"__main__\":\n    print(add(2, 3))\n"],
+        "output" : ["5"]
+        "result": {"metric_score": 0.0}
     }
 ]