source: PerfectFit/Dialog/src/test/java/SortedGoalsTest.java

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.io.IOException;

import org.junit.Test;

import tudelft.mentalhealth.perfectfit.Characteristics;
import tudelft.mentalhealth.perfectfit.GoalLine;
import tudelft.mentalhealth.perfectfit.SortedGoals;

/**
 * NOTE!! The ID value in the table is NOT equal to the row number It skips a
 * number of values for some reason. Eg there are no id 10 and 11.
 */
public class SortedGoalsTest {
        @Test
        public void test1() throws IOException {
                // copy of characteristics of example id=6 from the file
                Characteristics c6 = new Characteristics(1, 0.698430493273543,
                                0.604904632152589, 0.363636363636364, 0.75, 0.6,
                                0.285714285714286, 0.537037037037037, 0.25, 0.333333333333333);
                SortedGoals sorted = new SortedGoals(c6);

                checkSortedProperly(sorted, c6);
                // The firxt one will be one that is actually higher than c31
                assertEquals(6, sorted.get(1).getId());

        }

        @Test
        public void test31() throws IOException {
                // copy of characteristics of example 5 from the file
                Characteristics c31 = new Characteristics(1, 0.724215246636771,
                                0.272479564032698, 0.636363636363636, 0.25, 0.2,
                                0.357142857142857, 0.592592592592593, 0.333333333333333, 0.5);
                SortedGoals sorted = new SortedGoals(c31);

                checkSortedProperly(sorted, c31);
                // The firxt one will be one that is actually higher than c31
                assertEquals(31, sorted.get(1).getId());
        }

        @Test
        public void test43() throws IOException {
                // copy of characteristics of example 5 from the file
                Characteristics c43 = new Characteristics(1, 0.896860986547085,
                                0.574931880108992, 0.363636363636364, 0.75, 0.4, 0.5,
                                0.055555555555556, 0.25, 0.333333333333333);
                SortedGoals sorted = new SortedGoals(c43);

                checkSortedProperly(sorted, c43);
                // The firxt AND second one will be one that is actually higher than c31
                assertEquals(43, sorted.get(2).getId());

        }

        private void checkSortedProperly(SortedGoals sorted, Characteristics c) {
                GoalLine ex0 = sorted.get(0);
                double sim0 = ex0.characteristics().similarity(c);

                // check that the first one is indeed the best match
                for (GoalLine ex : sorted.getList()) {
                        assertTrue(
                                        "Example " + ex
                                                        + " is more similar to c31 than the top pick",
                                        ex.characteristics().similarity(c) <= sim0);
                }

        }
}