source: opponentmodel/src/test/java/geniusweb/profile/opponentmodel/FrequencyOppModelTest.java@ 14

package geniusweb.profile.opponentmodel;

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

import java.math.BigDecimal;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import org.junit.Test;

import geniusweb.issuevalue.Bid;
import geniusweb.issuevalue.DiscreteValue;
import geniusweb.issuevalue.DiscreteValueSet;
import geniusweb.issuevalue.Domain;
import geniusweb.issuevalue.NumberValue;
import geniusweb.issuevalue.NumberValueSet;
import geniusweb.issuevalue.Value;
import geniusweb.issuevalue.ValueSet;
import geniusweb.opponentmodel.FrequencyOpponentModel;
import tudelft.utilities.junit.GeneralTests;

public class FrequencyOppModelTest
                extends GeneralTests<FrequencyOpponentModel> {
        private final static String ISS1 = "issue1";
        private final static String ISS2 = "issue2";
        private static final DiscreteValue I1V1 = new DiscreteValue("i1v1");
        private static final DiscreteValue I1V2 = new DiscreteValue("i1v2");
        private static final DiscreteValue I1V2b = new DiscreteValue("i1v2b");

        private static Domain domain, domain2, domain3;
        private static FrequencyOpponentModel oppModel1, oppModel1b, oppModel2,
                        oppModel3, oppModel4;

        private final static List<List<Bid>> list = new LinkedList<>();

        private final static Bid bid1, bid2, bid3;
        private static final BigDecimal HALF = new BigDecimal("0.5");

        static {
                Map<String, ValueSet> issues = new HashMap<>();
                Collection<DiscreteValue> discretevalues1 = new LinkedList<>();
                discretevalues1.add(I1V1);
                discretevalues1.add(I1V2);
                DiscreteValueSet values1 = new DiscreteValueSet(discretevalues1);
                issues.put(ISS1, values1);
                NumberValueSet values2 = new NumberValueSet(BigDecimal.ZERO,
                                BigDecimal.TEN, new BigDecimal("0.3"));
                issues.put(ISS2, values2);
                domain = new Domain("test", issues);
                domain2 = new Domain("test2", issues);

                // slightly different issue1
                issues = new HashMap<>();
                discretevalues1 = new LinkedList<>();
                discretevalues1.add(I1V1);
                discretevalues1.add(I1V2b);
                values1 = new DiscreteValueSet(discretevalues1);
                issues.put(ISS1, values1);
                values2 = new NumberValueSet(BigDecimal.ZERO, BigDecimal.TEN,
                                new BigDecimal("0.3"));
                issues.put(ISS2, values2);
                domain3 = new Domain("test", issues);

                // all bids are for domain
                Map<String, Value> issuevalues = new HashMap<>();
                issuevalues.put(ISS1, I1V1);
                issuevalues.put(ISS2, new NumberValue(new BigDecimal("1.2")));
                bid1 = new Bid(issuevalues);

                issuevalues.put(ISS1, I1V1);
                issuevalues.put(ISS2, new NumberValue(new BigDecimal("1.5")));
                bid2 = new Bid(issuevalues);

                issuevalues.put(ISS1, I1V2);
                issuevalues.put(ISS2, new NumberValue(new BigDecimal("1.5")));
                bid3 = new Bid(issuevalues);

                oppModel1 = new FrequencyOpponentModel(domain);
                oppModel1b = new FrequencyOpponentModel(domain);
                oppModel2 = new FrequencyOpponentModel(domain2);
                oppModel3 = new FrequencyOpponentModel(domain3);
                oppModel4 = (FrequencyOpponentModel) oppModel3.update(bid1);

        }

        @Override
        public List<List<FrequencyOpponentModel>> getGeneralTestData() {
                return Arrays.asList(Arrays.asList(oppModel1, oppModel1b),
                                Arrays.asList(oppModel2), Arrays.asList(oppModel3),
                                Arrays.asList(oppModel4));
        }

        @Override
        public List<String> getGeneralTestStrings() {
                return Arrays.asList(
                                "FrequencyOpponentModel\\[0,\\{issue2=\\{\\}, issue1=\\{\\}\\}\\]",
                                "FrequencyOpponentModel\\[0,\\{issue2=\\{\\}, issue1=\\{\\}\\}\\]",
                                "FrequencyOpponentModel\\[0,\\{issue2=\\{\\}, issue1=\\{\\}\\}\\]",
                                "FrequencyOpponentModel\\[1,\\{issue2=\\{1.2=1\\}, issue1=\\{\"i1v1\"=1\\}\\}\\]");
        }

        @Test(expected = NullPointerException.class)
        public void smokeNullTest() {
                new FrequencyOpponentModel(null);
        }

        @Test
        public void smokeTest() {
                new FrequencyOpponentModel(domain);
        }

        @Test
        public void testEmptyModel() {
                FrequencyOpponentModel oppModel = new FrequencyOpponentModel(domain);
                assertEquals(BigDecimal.ONE, oppModel.getUtility(bid1));
                assertEquals(BigDecimal.ONE, oppModel.getUtility(bid2));
        }

        @Test
        public void testUpdate() {
                FrequencyOpponentModel oppModel = oppModel1.update(bid1);
                assertTrue(BigDecimal.ONE.compareTo(oppModel.getUtility(bid1)) == 0);
                assertTrue(BigDecimal.ZERO.compareTo(oppModel.getUtility(bid3)) == 0);
                // bid2 has 1 of 2 issue values same as bid1.
                assertTrue(HALF.compareTo(oppModel.getUtility(bid2)) == 0);
        }

        @Test
        public void testUpdate2() {
                // bid1 and bid2 both want I1V1. They differ on the number value.
                // bid3 wants I1V2 but does have the number value from bid2
                FrequencyOpponentModel oppModel = oppModel1.update(bid1).update(bid2);
                assertTrue(new BigDecimal("0.75")
                                .compareTo(oppModel.getUtility(bid1)) == 0);
                assertTrue(new BigDecimal("0.75")
                                .compareTo(oppModel.getUtility(bid2)) == 0);
                assertTrue(new BigDecimal("0.25")
                                .compareTo(oppModel.getUtility(bid3)) == 0);
        }

}