Adding (not completed) log of partial answers in Json.

3 files changed, 14 insertions, 17 deletions

diff --git a/test/uk/ac/ox/cs/pagoda/test_units/PagodaLUBM.java b/test/uk/ac/ox/cs/pagoda/test_units/PagodaLUBM.java
index 29abc87..1aded5b 100644
--- a/test/uk/ac/ox/cs/pagoda/test_units/PagodaLUBM.java
+++ b/test/uk/ac/ox/cs/pagoda/test_units/PagodaLUBM.java
@@ -13,7 +13,7 @@ public class PagodaLUBM {
                 TestGlobalCorrectness.test(Paths.get(ontoDir, "lubm/univ-bench.owl"),
                                                                    Paths.get(ontoDir, "lubm/data/lubm" + number + ".ttl"),
                                                                    Paths.get(ontoDir, "lubm/queries/test.sparql"),
-                                                                   Paths.get(ontoDir, "lubm/lubm" + number + ".ans"));
+                                                                   Paths.get(ontoDir, "lubm/lubm" + number + ".json"));
         }
 
         @Test
diff --git a/test/uk/ac/ox/cs/pagoda/test_units/PagodaUOBM.java b/test/uk/ac/ox/cs/pagoda/test_units/PagodaUOBM.java
index f53ed7e..ee9cf13 100644
--- a/test/uk/ac/ox/cs/pagoda/test_units/PagodaUOBM.java
+++ b/test/uk/ac/ox/cs/pagoda/test_units/PagodaUOBM.java
@@ -13,7 +13,7 @@ public class PagodaUOBM {
                 TestGlobalCorrectness.test(Paths.get(ontoDir, "uobm/univ-bench-dl.owl"),
                                                           Paths.get(ontoDir, "uobm/data/uobm" + number + ".ttl"),
                                                           Paths.get(ontoDir, "uobm/queries/test.sparql"),
-                                                          Paths.get(ontoDir, "uobm/uobm" + number + ".ans"));
+                                                          Paths.get(ontoDir, "uobm/uobm" + number + ".json"));
         }
 
         @Test
diff --git a/test/uk/ac/ox/cs/pagoda/test_units/TestGlobalCorrectness.java b/test/uk/ac/ox/cs/pagoda/test_units/TestGlobalCorrectness.java
index 34ec8cd..ffaf8fc 100644
--- a/test/uk/ac/ox/cs/pagoda/test_units/TestGlobalCorrectness.java
+++ b/test/uk/ac/ox/cs/pagoda/test_units/TestGlobalCorrectness.java
@@ -1,13 +1,12 @@
 package uk.ac.ox.cs.pagoda.test_units;
 
 import com.google.gson.Gson;
-import com.google.gson.GsonBuilder;
 import com.google.gson.reflect.TypeToken;
-import org.apache.log4j.Logger;
+import org.apache.log4j.Level;
 import org.testng.Assert;
-import uk.ac.ox.cs.pagoda.query.AnswerTuple;
 import uk.ac.ox.cs.pagoda.query.QueryRecord;
 import uk.ac.ox.cs.pagoda.tester.PagodaTester;
+import uk.ac.ox.cs.pagoda.util.Utility;
 
 import java.io.BufferedReader;
 import java.io.File;
@@ -16,39 +15,37 @@ import java.lang.reflect.Type;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
-import java.util.Collection;
 import java.util.Set;
 
-/*
+/**
  * This is a unit test for TestNG.
  * <p>
  * It tests the correctness on the final output.
  * */
 public class TestGlobalCorrectness {
 
-    public static final Logger logger = Logger.getLogger(TestGlobalCorrectness.class);
-
     public static void test(Path ontology, Path data, Path queries, Path givenAnswers) {
         try {
-            Path computedAnswers = Paths.get(File.createTempFile("answers", ".tmp").getAbsolutePath());
+            Utility.setLogLevel(Level.DEBUG);
+            Path computedAnswers = Paths.get(File.createTempFile("answers", ".json").getAbsolutePath());
             PagodaTester.main(ontology.toString(), data.toString(), queries.toString(), computedAnswers.toString());
-            Assert.assertTrue(checkSameContent(computedAnswers, givenAnswers));
+            assertSameContent(computedAnswers, givenAnswers);
         } catch (IOException e) {
             e.printStackTrace();
         }
     }
 
-    private static boolean checkSameContent(Path computedAnswersFile, Path givenAnswersFile) throws IOException {
+    private static void assertSameContent(Path computedAnswersFile, Path givenAnswersFile) throws IOException {
         BufferedReader computedReader = Files.newBufferedReader(computedAnswersFile);
         BufferedReader givenReader = Files.newBufferedReader(givenAnswersFile);
 
-        Gson gson = new GsonBuilder().create();
+        Gson gson = QueryRecord.GsonCreator.getInstance();
 
-        Type cqType = new TypeToken<Set<AnswerTuple>>() {}.getType();
-        Collection<QueryRecord> computedAnswers = gson.fromJson(computedReader, cqType);
-        Collection<QueryRecord> givenAnswers = gson.fromJson(givenReader, cqType);
+        Type cqType = new TypeToken<Set<QueryRecord>>() {}.getType();
+        Set<QueryRecord> computedAnswers = gson.fromJson(computedReader, cqType);
+        Set<QueryRecord> givenAnswers = gson.fromJson(givenReader, cqType);
 
-        return computedAnswers.equals(givenAnswers);
+        Assert.assertEquals(computedAnswers, givenAnswers);
     }