src/test/java/uk/ac/ox/cs/pagoda/query/CheckAnswers.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

package uk.ac.ox.cs.pagoda.query;

import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import org.testng.Assert;

import java.io.BufferedReader;
import java.io.IOException;
import java.lang.reflect.Type;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Set;

/**
 * It provides auxiliary methods for checking answers.
 */
public class CheckAnswers {

    private CheckAnswers() {
    }

    public static void assertSameAnswers(Path computedAnswersFile, Path givenAnswersFile) throws IOException {
        BufferedReader computedReader = Files.newBufferedReader(computedAnswersFile);
        BufferedReader givenReader = Files.newBufferedReader(givenAnswersFile);

        Gson gson = QueryRecord.GsonCreator.getInstance();

        Type cqType = new TypeToken<Set<QueryRecord>>() {}.getType();
        Set<QueryRecord> computedAnswersRecords = gson.fromJson(computedReader, cqType);
        Set<QueryRecord> givenAnswersRecords = gson.fromJson(givenReader, cqType);

        for(QueryRecord computedAnswersRecord : computedAnswersRecords) {
            Set<AnswerTuple> givenAnswers = null;
            for(QueryRecord givenAnswersRecord : givenAnswersRecords) {
                if(givenAnswersRecord.queryID == computedAnswersRecord.queryID) {
                    givenAnswers = givenAnswersRecord.soundAnswerTuples;
                    break;
                }
            }

            Assert.assertNotNull(givenAnswers, "Missing given answer for query no. " + computedAnswersRecord.queryID);

            Set<AnswerTuple> computedAnswers = computedAnswersRecord.soundAnswerTuples;
            Assert.assertEquals(computedAnswers.size(), givenAnswers.size(),
                                "Different number of sound answers for query " + computedAnswersRecord.queryID + "!"
                                        + "Expected " + givenAnswers.size() + ", got " + computedAnswers.size());
            Assert.assertEquals(computedAnswers, givenAnswers,
                                "Different sound answers for query " + computedAnswersRecord.queryID + "!");
        }
    }
}