Create tests.py

tests.py

@@ -0,0 +1,17 @@
+test_cases = [
+ {
+ "predictions": [0, 0],
+ "references": [1, 1],
+ "result": {"metric_score": 0}
+ },
+ {
+ "predictions": [1, 1],
+ "references": [1, 1],
+ "result": {"metric_score": 1}
+ },
+ {
+ "predictions": [1, 0],
+ "references": [1, 1],
+ "result": {"metric_score": 0.5}
+ }
+]