Ignore:
Timestamp:
08/16/12 14:45:03 (12 years ago)
Author:
sherbold
Message:
  • countless adaptations throughout nearly all components to remove errors introduced due to the refactoring of the event core
Location:
trunk/quest-core-usageprofiles-test/src/test/java/de/ugoe/cs/quest/usageprofiles
Files:
2 added
7 edited

Legend:

Unmodified
Added
Removed

  • trunk/quest-core-usageprofiles-test/src/test/java/de/ugoe/cs/quest/usageprofiles/DeterministicFiniteAutomatonTest.java

    r518 r548  
    77import de.ugoe.cs.quest.eventcore.Event; 
    88import de.ugoe.cs.quest.usageprofiles.DeterministicFiniteAutomaton; 
     9import de.ugoe.cs.quest.usageprofiles.mock.StringEventType; 
    910 
    1011import java.util.Random; 
    … …  
    2223public class DeterministicFiniteAutomatonTest { 
    2324 
    24         Collection<List<? extends Event<?>>> sequences; 
     25        Collection<List<Event>> sequences; 
    2526 
    2627        @Test 
    … …  
    4647                fixture.train(sequences); 
    4748 
    48                 List<Event<String>> context = new ArrayList<Event<String>>(); 
    49                 context.add(new Event<String>("a")); 
    50                 context.add(new Event<String>("b")); 
     49                List<Event> context = new ArrayList<Event>(); 
     50                context.add(new Event(new StringEventType("a"))); 
     51                context.add(new Event(new StringEventType("b"))); 
    5152 
    52                 Event<String> symbol = new Event<String>("r"); 
     53                Event symbol = new Event(new StringEventType("r")); 
    5354 
    5455                double result = fixture.getProbability(context, symbol); 
    … …  
    6364                fixture.train(sequences); 
    6465 
    65                 List<Event<String>> context = new ArrayList<Event<String>>(); 
    66                 context.add(new Event<String>("a")); 
     66                List<Event> context = new ArrayList<Event>(); 
     67                context.add(new Event(new StringEventType("a"))); 
    6768 
    68                 Event<String> symbol = new Event<String>("b"); 
     69                Event symbol = new Event(new StringEventType("b")); 
    6970 
    7071                double result = fixture.getProbability(context, symbol); 
    … …  
    7980                fixture.train(sequences); 
    8081 
    81                 List<Event<String>> context = new ArrayList<Event<String>>(); 
    82                 context.add(new Event<String>("a")); 
     82                List<Event> context = new ArrayList<Event>(); 
     83                context.add(new Event(new StringEventType("a"))); 
    8384 
    84                 Event<String> symbol = new Event<String>("c"); 
     85                Event symbol = new Event(new StringEventType("c")); 
    8586 
    8687                double result = fixture.getProbability(context, symbol); 
    … …  
    9596                fixture.train(sequences); 
    9697 
    97                 List<Event<String>> context = new ArrayList<Event<String>>(); 
    98                 context.add(new Event<String>("a")); 
     98                List<Event> context = new ArrayList<Event>(); 
     99                context.add(new Event(new StringEventType("a"))); 
    99100 
    100                 Event<String> symbol = new Event<String>("e"); 
     101                Event symbol = new Event(new StringEventType("e")); 
    101102 
    102103                double result = fixture.getProbability(context, symbol); 
    … …  
    111112                fixture.train(sequences); 
    112113 
    113                 List<Event<String>> context = new ArrayList<Event<String>>(); 
    114                 context.add(new Event<String>("a")); 
     114                List<Event> context = new ArrayList<Event>(); 
     115                context.add(new Event(new StringEventType("a"))); 
    115116 
    116                 Event<String> symbol = null; 
     117                Event symbol = null; 
    117118 
    118119                fixture.getProbability(context, symbol); 
    … …  
    125126                fixture.train(sequences); 
    126127 
    127                 List<Event<String>> context = null; 
     128                List<Event> context = null; 
    128129 
    129                 Event<String> symbol = new Event<String>("a"); 
     130                Event symbol = new Event(new StringEventType("a")); 
    130131 
    131132                fixture.getProbability(context, symbol); 
    … …  
    134135        @Before 
    135136        public void setUp() throws Exception { 
    136                 List<Event<?>> sequence = new ArrayList<Event<?>>(); 
    137                 sequence.add(new Event<String>("a")); 
    138                 sequence.add(new Event<String>("b")); 
    139                 sequence.add(new Event<String>("r")); 
    140                 sequence.add(new Event<String>("a")); 
    141                 sequence.add(new Event<String>("c")); 
    142                 sequence.add(new Event<String>("a")); 
    143                 sequence.add(new Event<String>("d")); 
    144                 sequence.add(new Event<String>("a")); 
    145                 sequence.add(new Event<String>("b")); 
    146                 sequence.add(new Event<String>("r")); 
    147                 sequence.add(new Event<String>("a")); 
     137                List<Event> sequence = new ArrayList<Event>(); 
     138                sequence.add(new Event(new StringEventType("a"))); 
     139                sequence.add(new Event(new StringEventType("b"))); 
     140                sequence.add(new Event(new StringEventType("r"))); 
     141                sequence.add(new Event(new StringEventType("a"))); 
     142                sequence.add(new Event(new StringEventType("c"))); 
     143                sequence.add(new Event(new StringEventType("a"))); 
     144                sequence.add(new Event(new StringEventType("d"))); 
     145                sequence.add(new Event(new StringEventType("a"))); 
     146                sequence.add(new Event(new StringEventType("b"))); 
     147                sequence.add(new Event(new StringEventType("r"))); 
     148                sequence.add(new Event(new StringEventType("a"))); 
    148149 
    149                 sequences = new ArrayList<List<? extends Event<?>>>(); 
     150                sequences = new ArrayList<List<Event>>(); 
    150151                sequences.add(sequence); 
    151152        } 

  • trunk/quest-core-usageprofiles-test/src/test/java/de/ugoe/cs/quest/usageprofiles/FirstOrderMarkovModelTest.java

    r518 r548  
    1010import de.ugoe.cs.quest.usageprofiles.FirstOrderMarkovModel; 
    1111import de.ugoe.cs.quest.usageprofiles.FirstOrderMarkovModel.MarkovEdge; 
     12import de.ugoe.cs.quest.usageprofiles.mock.StringEventType; 
    1213import static org.junit.Assert.*; 
    1314 
    … …  
    2122public class FirstOrderMarkovModelTest { 
    2223 
    23         Collection<List<? extends Event<?>>> sequences; 
     24        Collection<List<Event>> sequences; 
    2425         
    2526        @Test 
    … …  
    7273        @Before 
    7374        public void setUp() throws Exception { 
    74                 List<Event<?>> sequence = new ArrayList<Event<?>>(); 
    75                 sequence.add(new Event<String>("a")); 
    76                 sequence.add(new Event<String>("b")); 
    77                 sequence.add(new Event<String>("r")); 
    78                 sequence.add(new Event<String>("a")); 
    79                 sequence.add(new Event<String>("c")); 
    80                 sequence.add(new Event<String>("a")); 
    81                 sequence.add(new Event<String>("d")); 
    82                 sequence.add(new Event<String>("a")); 
    83                 sequence.add(new Event<String>("b")); 
    84                 sequence.add(new Event<String>("r")); 
    85                 sequence.add(new Event<String>("a")); 
     75                List<Event> sequence = new ArrayList<Event>(); 
     76                sequence.add(new Event(new StringEventType("a"))); 
     77                sequence.add(new Event(new StringEventType("b"))); 
     78                sequence.add(new Event(new StringEventType("r"))); 
     79                sequence.add(new Event(new StringEventType("a"))); 
     80                sequence.add(new Event(new StringEventType("c"))); 
     81                sequence.add(new Event(new StringEventType("a"))); 
     82                sequence.add(new Event(new StringEventType("d"))); 
     83                sequence.add(new Event(new StringEventType("a"))); 
     84                sequence.add(new Event(new StringEventType("b"))); 
     85                sequence.add(new Event(new StringEventType("r"))); 
     86                sequence.add(new Event(new StringEventType("a"))); 
    8687 
    87                 sequences = new ArrayList<List<? extends Event<?>>>(); 
     88                sequences = new ArrayList<List<Event>>(); 
    8889                sequences.add(sequence); 
    8990        } 

  • trunk/quest-core-usageprofiles-test/src/test/java/de/ugoe/cs/quest/usageprofiles/HighOrderMarkovModelTest.java

    r518 r548  
    77import de.ugoe.cs.quest.eventcore.Event; 
    88import de.ugoe.cs.quest.usageprofiles.HighOrderMarkovModel; 
     9import de.ugoe.cs.quest.usageprofiles.mock.StringEventType; 
    910 
    1011import java.util.Random; 
    … …  
    2223public class HighOrderMarkovModelTest { 
    2324 
    24         Collection<List<? extends Event<?>>> sequences; 
     25        Collection<List<Event>> sequences; 
    2526 
    2627        @Test 
    … …  
    7172                fixture.train(sequences); 
    7273 
    73                 List<Event<String>> context = new ArrayList<Event<String>>(); 
    74                 context.add(new Event<String>("a")); 
    75  
    76                 Event<String> symbol = new Event<String>("b"); 
     74                List<Event> context = new ArrayList<Event>(); 
     75                context.add(new Event(new StringEventType("a"))); 
     76 
     77                Event symbol = new Event(new StringEventType("b")); 
    7778 
    7879                double result = fixture.getProbability(context, symbol); 
    … …  
    8889                fixture.train(sequences); 
    8990 
    90                 List<Event<String>> context = new ArrayList<Event<String>>(); 
    91                 context.add(new Event<String>("a")); 
    92  
    93                 Event<String> symbol = new Event<String>("r"); 
     91                List<Event> context = new ArrayList<Event>(); 
     92                context.add(new Event(new StringEventType("a"))); 
     93 
     94                Event symbol = new Event(new StringEventType("r")); 
    9495 
    9596                double result = fixture.getProbability(context, symbol); 
    … …  
    105106                fixture.train(sequences); 
    106107 
    107                 List<Event<String>> context = new ArrayList<Event<String>>(); 
    108                 context.add(new Event<String>("a")); 
    109  
    110                 Event<String> symbol = new Event<String>("c"); 
     108                List<Event> context = new ArrayList<Event>(); 
     109                context.add(new Event(new StringEventType("a"))); 
     110 
     111                Event symbol = new Event(new StringEventType("c")); 
    111112 
    112113                double result = fixture.getProbability(context, symbol); 
    … …  
    122123                fixture.train(sequences); 
    123124 
    124                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    125                 context.add(Event.STARTEVENT); 
    126                 context.add(new Event<String>("a")); 
    127  
    128                 Event<String> symbol = new Event<String>("b"); 
     125                List<Event> context = new ArrayList<Event>(); 
     126                context.add(Event.STARTEVENT); 
     127                context.add(new Event(new StringEventType("a"))); 
     128 
     129                Event symbol = new Event(new StringEventType("b")); 
    129130 
    130131                double result = fixture.getProbability(context, symbol); 
    … …  
    140141                fixture.train(sequences); 
    141142 
    142                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    143                 context.add(Event.STARTEVENT); 
    144                 context.add(new Event<String>("a")); 
    145  
    146                 Event<String> symbol = new Event<String>("b"); 
     143                List<Event> context = new ArrayList<Event>(); 
     144                context.add(Event.STARTEVENT); 
     145                context.add(new Event(new StringEventType("a"))); 
     146 
     147                Event symbol = new Event(new StringEventType("b")); 
    147148 
    148149                double result = fixture.getProbability(context, symbol); 
    … …  
    158159                fixture.train(sequences); 
    159160 
    160                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    161                 context.add(Event.STARTEVENT); 
    162                 context.add(new Event<String>("b")); 
    163  
    164                 Event<String> symbol = new Event<String>("b"); 
     161                List<Event> context = new ArrayList<Event>(); 
     162                context.add(Event.STARTEVENT); 
     163                context.add(new Event(new StringEventType("b"))); 
     164 
     165                Event symbol = new Event(new StringEventType("b")); 
    165166 
    166167                double result = fixture.getProbability(context, symbol); 
    … …  
    176177                fixture.train(sequences); 
    177178 
    178                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    179                 context.add(Event.STARTEVENT); 
    180                 context.add(new Event<String>("b")); 
    181  
    182                 Event<String> symbol = new Event<String>("a"); 
     179                List<Event> context = new ArrayList<Event>(); 
     180                context.add(Event.STARTEVENT); 
     181                context.add(new Event(new StringEventType("b"))); 
     182 
     183                Event symbol = new Event(new StringEventType("a")); 
    183184 
    184185                double result = fixture.getProbability(context, symbol); 
    … …  
    194195                fixture.train(sequences); 
    195196 
    196                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    197                 context.add(Event.STARTEVENT); 
    198                 context.add(new Event<String>("b")); 
    199  
    200                 Event<String> symbol = null; 
     197                List<Event> context = new ArrayList<Event>(); 
     198                context.add(Event.STARTEVENT); 
     199                context.add(new Event(new StringEventType("b"))); 
     200 
     201                Event symbol = null; 
    201202 
    202203                fixture.getProbability(context, symbol); 
    … …  
    210211                fixture.train(sequences); 
    211212 
    212                 List<Event<?>> context = null; 
    213  
    214                 Event<String> symbol = new Event<String>("b"); 
     213                List<Event> context = null; 
     214 
     215                Event symbol = new Event(new StringEventType("b")); 
    215216 
    216217                fixture.getProbability(context, symbol); 
    … …  
    219220        @Before 
    220221        public void setUp() throws Exception { 
    221                 List<Event<?>> sequence = new ArrayList<Event<?>>(); 
    222                 sequence.add(new Event<String>("a")); 
    223                 sequence.add(new Event<String>("b")); 
    224                 sequence.add(new Event<String>("r")); 
    225                 sequence.add(new Event<String>("a")); 
    226                 sequence.add(new Event<String>("c")); 
    227                 sequence.add(new Event<String>("a")); 
    228                 sequence.add(new Event<String>("d")); 
    229                 sequence.add(new Event<String>("a")); 
    230                 sequence.add(new Event<String>("b")); 
    231                 sequence.add(new Event<String>("r")); 
    232                 sequence.add(new Event<String>("a")); 
    233  
    234                 sequences = new ArrayList<List<? extends Event<?>>>(); 
     222                List<Event> sequence = new ArrayList<Event>(); 
     223                sequence.add(new Event(new StringEventType("a"))); 
     224                sequence.add(new Event(new StringEventType("b"))); 
     225                sequence.add(new Event(new StringEventType("r"))); 
     226                sequence.add(new Event(new StringEventType("a"))); 
     227                sequence.add(new Event(new StringEventType("c"))); 
     228                sequence.add(new Event(new StringEventType("a"))); 
     229                sequence.add(new Event(new StringEventType("d"))); 
     230                sequence.add(new Event(new StringEventType("a"))); 
     231                sequence.add(new Event(new StringEventType("b"))); 
     232                sequence.add(new Event(new StringEventType("r"))); 
     233                sequence.add(new Event(new StringEventType("a"))); 
     234 
     235                sequences = new ArrayList<List<Event>>(); 
    235236                sequences.add(sequence); 
    236237        } 

  • trunk/quest-core-usageprofiles-test/src/test/java/de/ugoe/cs/quest/usageprofiles/MockTrieBasedModel.java

    r518 r548  
    1616 
    1717        @Override 
    18         public double getProbability(List<? extends Event<?>> context, 
    19                         Event<?> symbol) { 
    20                 List<Event<?>> list = new ArrayList<Event<?>>(); 
     18        public double getProbability(List<Event> context, 
     19                        Event symbol) { 
     20                List<Event> list = new ArrayList<Event>(); 
    2121                if( context.isEmpty() ) { 
    2222                        return 2; 

  • trunk/quest-core-usageprofiles-test/src/test/java/de/ugoe/cs/quest/usageprofiles/ModelFlattenerTest.java

    r518 r548  
     1 
    12package de.ugoe.cs.quest.usageprofiles; 
    23 
    … …  
    1415import de.ugoe.cs.quest.usageprofiles.PredictionByPartialMatch; 
    1516import de.ugoe.cs.quest.usageprofiles.TrieNode; 
     17import de.ugoe.cs.quest.usageprofiles.mock.StringEventType; 
    1618import static org.junit.Assert.*; 
    1719 
    1820/** 
    19  * The class <code>ModelFlattenerTest</code> contains tests for the class <code>{@link ModelFlattener}</code>. 
    20  * 
     21 * The class <code>ModelFlattenerTest</code> contains tests for the class 
     22 * <code>{@link ModelFlattener}</code>. 
     23 *  
    2124 * @author Steffen Herbold 
    2225 * @version 1.0 
    2326 */ 
    2427public class ModelFlattenerTest { 
    25          
    26         List<Event<?>> sequence; 
    27          
    28         private static void assertCollectionContent(Collection<?> c1, Collection<?> c2) { 
    29                 assertEquals(c1.size(), c2.size()); 
    30                 for( Object obj : c1 ) { 
    31                         assertTrue(c2.contains(obj)); 
    32                 } 
    33         } 
    34          
    35         @Test 
    36         public void testFlattenHighOrderMarkovModel_1() 
    37                 throws Exception { 
    38                 ModelFlattener fixture = new ModelFlattener(); 
    39                 HighOrderMarkovModel model = new HighOrderMarkovModel(2, new Random()); 
    40                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
    41                 sequences.add(sequence); 
    42                 model.train(sequences); 
    43                  
    44                 Collection<Event<?>> expectedSymbols = new HashSet<Event<?>>(); 
    45                 expectedSymbols.add(new Event<Object>("a-=-END")); 
    46                 expectedSymbols.add(new Event<Object>("a-=-b")); 
    47                 expectedSymbols.add(new Event<Object>("a-=-c")); 
    48                 expectedSymbols.add(new Event<Object>("a-=-d")); 
    49                 expectedSymbols.add(new Event<Object>("b-=-r")); 
    50                 expectedSymbols.add(new Event<Object>("c-=-a")); 
    51                 expectedSymbols.add(new Event<Object>("d-=-a")); 
    52                 expectedSymbols.add(new Event<Object>("r-=-a")); 
    53                 expectedSymbols.add(new Event<Object>("START-=-a")); 
    5428 
    55                 FirstOrderMarkovModel result = fixture.flattenHighOrderMarkovModel(model); 
    56                  
    57                 assertCollectionContent(expectedSymbols, result.getEvents()); 
    58                  
    59                 TrieNode<Event<?>> root = result.trie.find(null); 
    60                 TrieNode<Event<?>> root_aEnd = root.getChild(new Event<Object>("a-=-END")); 
    61                 TrieNode<Event<?>> root_ab = root.getChild(new Event<Object>("a-=-b")); 
    62                 TrieNode<Event<?>> root_ab_br = root_ab.getChild(new Event<Object>("b-=-r")); 
    63                 TrieNode<Event<?>> root_ac = root.getChild(new Event<Object>("a-=-c")); 
    64                 TrieNode<Event<?>> root_ac_ca = root_ac.getChild(new Event<Object>("c-=-a")); 
    65                 TrieNode<Event<?>> root_ad = root.getChild(new Event<Object>("a-=-d")); 
    66                 TrieNode<Event<?>> root_ad_da = root_ad.getChild(new Event<Object>("d-=-a")); 
    67                 TrieNode<Event<?>> root_br = root.getChild(new Event<Object>("b-=-r")); 
    68                 TrieNode<Event<?>> root_br_ra = root_br.getChild(new Event<Object>("r-=-a")); 
    69                 TrieNode<Event<?>> root_ca = root.getChild(new Event<Object>("c-=-a")); 
    70                 TrieNode<Event<?>> root_ca_ad = root_ca.getChild(new Event<Object>("a-=-d")); 
    71                 TrieNode<Event<?>> root_da = root.getChild(new Event<Object>("d-=-a")); 
    72                 TrieNode<Event<?>> root_da_ab = root_da.getChild(new Event<Object>("a-=-b")); 
    73                 TrieNode<Event<?>> root_ra = root.getChild(new Event<Object>("r-=-a")); 
    74                 TrieNode<Event<?>> root_ra_ac = root_ra.getChild(new Event<Object> ("a-=-c")); 
    75                 TrieNode<Event<?>> root_ra_aEnd = root_ra.getChild(new Event<Object>("a-=-END")); 
    76                 TrieNode<Event<?>> root_startA = root.getChild(new Event<Object>("START-=-a")); 
    77                 TrieNode<Event<?>> root_startA_ab = root_startA.getChild(new Event<Object>("a-=-b")); 
    78                  
    79                 assertEquals(1, root_aEnd.getCount()); 
    80                 assertTrue(root_aEnd.isLeaf()); 
    81                 assertEquals(2, root_ab.getCount()); 
    82                 assertEquals(1, root_ab.getChildren().size()); 
    83                 assertEquals(2, root_ab_br.getCount()); 
    84                 assertTrue(root_ab_br.isLeaf()); 
    85                 assertEquals(1, root_ac.getCount()); 
    86                 assertEquals(1, root_ac.getChildren().size()); 
    87                 assertEquals(1, root_ac_ca.getCount()); 
    88                 assertTrue(root_ac_ca.isLeaf()); 
    89                 assertEquals(1, root_ad.getCount()); 
    90                 assertEquals(1, root_ad.getChildren().size()); 
    91                 assertEquals(1, root_ad_da.getCount()); 
    92                 assertTrue(root_ad_da.isLeaf()); 
    93                 assertEquals(2, root_br.getCount()); 
    94                 assertEquals(1, root_br.getChildren().size()); 
    95                 assertEquals(2, root_br_ra.getCount()); 
    96                 assertTrue(root_br_ra.isLeaf()); 
    97                 assertEquals(1, root_ca.getCount()); 
    98                 assertEquals(1, root_ca.getChildren().size()); 
    99                 assertEquals(1, root_ca_ad.getCount()); 
    100                 assertTrue(root_ca_ad.isLeaf()); 
    101                 assertEquals(1, root_da.getCount()); 
    102                 assertEquals(1, root_da.getChildren().size()); 
    103                 assertEquals(1, root_da_ab.getCount()); 
    104                 assertTrue(root_da_ab.isLeaf()); 
    105                 assertEquals(2, root_ra.getCount()); 
    106                 assertEquals(2, root_ra.getChildren().size()); 
    107                 assertEquals(1, root_ra_ac.getCount()); 
    108                 assertTrue(root_ra_ac.isLeaf()); 
    109                 assertEquals(1, root_ra_aEnd.getCount()); 
    110                 assertTrue(root_ra_aEnd.isLeaf()); 
    111                 assertEquals(1, root_startA.getCount()); 
    112                 assertEquals(1, root_startA.getChildren().size()); 
    113                 assertEquals(1, root_startA_ab.getCount()); 
    114                 assertTrue(root_startA_ab.isLeaf());             
    115         } 
     29    List<Event> sequence; 
    11630 
    117         @Test 
    118         public void testFlattenPredictionByPartialMatch_1() 
    119                 throws Exception { 
    120                 ModelFlattener fixture = new ModelFlattener(); 
    121                 PredictionByPartialMatch model = new PredictionByPartialMatch(1, new Random()); 
     31    private static void assertCollectionContent(Collection<?> c1, Collection<?> c2) { 
     32        assertEquals(c1.size(), c2.size()); 
     33        for (Object obj : c1) { 
     34            assertTrue(c2.contains(obj)); 
     35        } 
     36    } 
    12237 
    123                 FirstOrderMarkovModel result = fixture.flattenPredictionByPartialMatch(model); 
    124                  
    125                 assertEquals(null, result); 
    126         } 
     38    @Test 
     39    public void testFlattenHighOrderMarkovModel_1() throws Exception { 
     40        ModelFlattener fixture = new ModelFlattener(); 
     41        HighOrderMarkovModel model = new HighOrderMarkovModel(2, new Random()); 
     42        Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
     43        sequences.add(sequence); 
     44        model.train(sequences); 
    12745 
    128         @Before 
    129         public void setUp() 
    130                 throws Exception { 
    131                 sequence = new ArrayList<Event<?>>(); 
    132                 sequence.add(new Event<String>("a")); 
    133                 sequence.add(new Event<String>("b")); 
    134                 sequence.add(new Event<String>("r")); 
    135                 sequence.add(new Event<String>("a")); 
    136                 sequence.add(new Event<String>("c")); 
    137                 sequence.add(new Event<String>("a")); 
    138                 sequence.add(new Event<String>("d")); 
    139                 sequence.add(new Event<String>("a")); 
    140                 sequence.add(new Event<String>("b")); 
    141                 sequence.add(new Event<String>("r")); 
    142                 sequence.add(new Event<String>("a")); 
    143         } 
     46        Collection<Event> expectedSymbols = new HashSet<Event>(); 
     47        expectedSymbols.add(new Event(new StringEventType("a-=-END"))); 
     48        expectedSymbols.add(new Event(new StringEventType("a-=-b"))); 
     49        expectedSymbols.add(new Event(new StringEventType("a-=-c"))); 
     50        expectedSymbols.add(new Event(new StringEventType("a-=-d"))); 
     51        expectedSymbols.add(new Event(new StringEventType("b-=-r"))); 
     52        expectedSymbols.add(new Event(new StringEventType("c-=-a"))); 
     53        expectedSymbols.add(new Event(new StringEventType("d-=-a"))); 
     54        expectedSymbols.add(new Event(new StringEventType("r-=-a"))); 
     55        expectedSymbols.add(new Event(new StringEventType("START-=-a"))); 
    14456 
    145         public static void main(String[] args) { 
    146                 new org.junit.runner.JUnitCore().run(ModelFlattenerTest.class); 
    147         } 
     57        FirstOrderMarkovModel result = fixture.flattenHighOrderMarkovModel(model); 
     58 
     59        assertCollectionContent(expectedSymbols, result.getEvents()); 
     60 
     61        TrieNode<Event> root = result.trie.find(null); 
     62        TrieNode<Event> root_aEnd = root.getChild(new Event(new StringEventType("a-=-END"))); 
     63        TrieNode<Event> root_ab = root.getChild(new Event(new StringEventType("a-=-b"))); 
     64        TrieNode<Event> root_ab_br = root_ab.getChild(new Event(new StringEventType("b-=-r"))); 
     65        TrieNode<Event> root_ac = root.getChild(new Event(new StringEventType("a-=-c"))); 
     66        TrieNode<Event> root_ac_ca = root_ac.getChild(new Event(new StringEventType("c-=-a"))); 
     67        TrieNode<Event> root_ad = root.getChild(new Event(new StringEventType("a-=-d"))); 
     68        TrieNode<Event> root_ad_da = root_ad.getChild(new Event(new StringEventType("d-=-a"))); 
     69        TrieNode<Event> root_br = root.getChild(new Event(new StringEventType("b-=-r"))); 
     70        TrieNode<Event> root_br_ra = root_br.getChild(new Event(new StringEventType("r-=-a"))); 
     71        TrieNode<Event> root_ca = root.getChild(new Event(new StringEventType("c-=-a"))); 
     72        TrieNode<Event> root_ca_ad = root_ca.getChild(new Event(new StringEventType("a-=-d"))); 
     73        TrieNode<Event> root_da = root.getChild(new Event(new StringEventType("d-=-a"))); 
     74        TrieNode<Event> root_da_ab = root_da.getChild(new Event(new StringEventType("a-=-b"))); 
     75        TrieNode<Event> root_ra = root.getChild(new Event(new StringEventType("r-=-a"))); 
     76        TrieNode<Event> root_ra_ac = root_ra.getChild(new Event(new StringEventType("a-=-c"))); 
     77        TrieNode<Event> root_ra_aEnd = root_ra.getChild(new Event(new StringEventType("a-=-END"))); 
     78        TrieNode<Event> root_startA = root.getChild(new Event(new StringEventType("START-=-a"))); 
     79        TrieNode<Event> root_startA_ab = 
     80            root_startA.getChild(new Event(new StringEventType("a-=-b"))); 
     81 
     82        assertEquals(1, root_aEnd.getCount()); 
     83        assertTrue(root_aEnd.isLeaf()); 
     84        assertEquals(2, root_ab.getCount()); 
     85        assertEquals(1, root_ab.getChildren().size()); 
     86        assertEquals(2, root_ab_br.getCount()); 
     87        assertTrue(root_ab_br.isLeaf()); 
     88        assertEquals(1, root_ac.getCount()); 
     89        assertEquals(1, root_ac.getChildren().size()); 
     90        assertEquals(1, root_ac_ca.getCount()); 
     91        assertTrue(root_ac_ca.isLeaf()); 
     92        assertEquals(1, root_ad.getCount()); 
     93        assertEquals(1, root_ad.getChildren().size()); 
     94        assertEquals(1, root_ad_da.getCount()); 
     95        assertTrue(root_ad_da.isLeaf()); 
     96        assertEquals(2, root_br.getCount()); 
     97        assertEquals(1, root_br.getChildren().size()); 
     98        assertEquals(2, root_br_ra.getCount()); 
     99        assertTrue(root_br_ra.isLeaf()); 
     100        assertEquals(1, root_ca.getCount()); 
     101        assertEquals(1, root_ca.getChildren().size()); 
     102        assertEquals(1, root_ca_ad.getCount()); 
     103        assertTrue(root_ca_ad.isLeaf()); 
     104        assertEquals(1, root_da.getCount()); 
     105        assertEquals(1, root_da.getChildren().size()); 
     106        assertEquals(1, root_da_ab.getCount()); 
     107        assertTrue(root_da_ab.isLeaf()); 
     108        assertEquals(2, root_ra.getCount()); 
     109        assertEquals(2, root_ra.getChildren().size()); 
     110        assertEquals(1, root_ra_ac.getCount()); 
     111        assertTrue(root_ra_ac.isLeaf()); 
     112        assertEquals(1, root_ra_aEnd.getCount()); 
     113        assertTrue(root_ra_aEnd.isLeaf()); 
     114        assertEquals(1, root_startA.getCount()); 
     115        assertEquals(1, root_startA.getChildren().size()); 
     116        assertEquals(1, root_startA_ab.getCount()); 
     117        assertTrue(root_startA_ab.isLeaf()); 
     118    } 
     119 
     120    @Test 
     121    public void testFlattenPredictionByPartialMatch_1() throws Exception { 
     122        ModelFlattener fixture = new ModelFlattener(); 
     123        PredictionByPartialMatch model = new PredictionByPartialMatch(1, new Random()); 
     124 
     125        FirstOrderMarkovModel result = fixture.flattenPredictionByPartialMatch(model); 
     126 
     127        assertEquals(null, result); 
     128    } 
     129 
     130    @Before 
     131    public void setUp() throws Exception { 
     132        sequence = new ArrayList<Event>(); 
     133        sequence.add(new Event(new StringEventType("a"))); 
     134        sequence.add(new Event(new StringEventType("b"))); 
     135        sequence.add(new Event(new StringEventType("r"))); 
     136        sequence.add(new Event(new StringEventType("a"))); 
     137        sequence.add(new Event(new StringEventType("c"))); 
     138        sequence.add(new Event(new StringEventType("a"))); 
     139        sequence.add(new Event(new StringEventType("d"))); 
     140        sequence.add(new Event(new StringEventType("a"))); 
     141        sequence.add(new Event(new StringEventType("b"))); 
     142        sequence.add(new Event(new StringEventType("r"))); 
     143        sequence.add(new Event(new StringEventType("a"))); 
     144    } 
     145 
     146    public static void main(String[] args) { 
     147        new org.junit.runner.JUnitCore().run(ModelFlattenerTest.class); 
     148    } 
    148149} 

  • trunk/quest-core-usageprofiles-test/src/test/java/de/ugoe/cs/quest/usageprofiles/PredictionByPartialMatchTest.java

    r518 r548  
    77import de.ugoe.cs.quest.eventcore.Event; 
    88import de.ugoe.cs.quest.usageprofiles.PredictionByPartialMatch; 
     9import de.ugoe.cs.quest.usageprofiles.mock.StringEventType; 
    910 
    1011import java.util.Random; 
    … …  
    2223public class PredictionByPartialMatchTest { 
    2324 
    24         Collection<List<? extends Event<?>>> sequences; 
     25        Collection<List<Event>> sequences; 
    2526 
    2627        @Test 
    … …  
    209210                fixture.train(sequences); 
    210211                 
    211                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    212                 context.add(Event.STARTEVENT); 
    213                 context.add(new Event<String>("a")); 
    214  
    215                 Event<String> symbol = new Event<String>("b"); 
     212                List<Event> context = new ArrayList<Event>(); 
     213                context.add(Event.STARTEVENT); 
     214                context.add(new Event(new StringEventType("a"))); 
     215 
     216                Event symbol = new Event(new StringEventType("b")); 
    216217                 
    217218                double result = fixture.getProbability(context, symbol); 
    … …  
    230231                fixture.train(sequences); 
    231232                 
    232                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    233                 context.add(Event.STARTEVENT); 
    234                 context.add(new Event<String>("a")); 
    235  
    236                 Event<String> symbol = new Event<String>("c"); 
     233                List<Event> context = new ArrayList<Event>(); 
     234                context.add(Event.STARTEVENT); 
     235                context.add(new Event(new StringEventType("a"))); 
     236 
     237                Event symbol = new Event(new StringEventType("c")); 
    237238                 
    238239                double result = fixture.getProbability(context, symbol); 
    … …  
    251252                fixture.train(sequences); 
    252253                 
    253                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    254                 context.add(Event.STARTEVENT); 
    255                 context.add(new Event<String>("a")); 
    256  
    257                 Event<String> symbol = new Event<String>("b"); 
     254                List<Event> context = new ArrayList<Event>(); 
     255                context.add(Event.STARTEVENT); 
     256                context.add(new Event(new StringEventType("a"))); 
     257 
     258                Event symbol = new Event(new StringEventType("b")); 
    258259                 
    259260                double result = fixture.getProbability(context, symbol); 
    … …  
    272273                fixture.train(sequences); 
    273274                 
    274                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    275                 context.add(Event.STARTEVENT); 
    276                 context.add(new Event<String>("a")); 
    277  
    278                 Event<String> symbol = new Event<String>("c"); 
     275                List<Event> context = new ArrayList<Event>(); 
     276                context.add(Event.STARTEVENT); 
     277                context.add(new Event(new StringEventType("a"))); 
     278 
     279                Event symbol = new Event(new StringEventType("c")); 
    279280                 
    280281                double result = fixture.getProbability(context, symbol); 
    … …  
    293294                fixture.train(sequences); 
    294295                 
    295                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    296                 context.add(Event.STARTEVENT); 
    297                 context.add(new Event<String>("a")); 
    298  
    299                 Event<String> symbol = new Event<String>("b"); 
     296                List<Event> context = new ArrayList<Event>(); 
     297                context.add(Event.STARTEVENT); 
     298                context.add(new Event(new StringEventType("a"))); 
     299 
     300                Event symbol = new Event(new StringEventType("b")); 
    300301                 
    301302                double result = fixture.getProbability(context, symbol); 
    … …  
    314315                fixture.train(sequences); 
    315316                 
    316                 List<Event<?>> context = new ArrayList<Event<?>>(); 
    317                 context.add(Event.STARTEVENT); 
    318                 context.add(new Event<String>("a")); 
    319  
    320                 Event<String> symbol = new Event<String>("c"); 
     317                List<Event> context = new ArrayList<Event>(); 
     318                context.add(Event.STARTEVENT); 
     319                context.add(new Event(new StringEventType("a"))); 
     320 
     321                Event symbol = new Event(new StringEventType("c")); 
    321322                 
    322323                double result = fixture.getProbability(context, symbol); 
    … …  
    342343        @Before 
    343344        public void setUp() throws Exception { 
    344                 List<Event<?>> sequence = new ArrayList<Event<?>>(); 
    345                 sequence.add(new Event<String>("a")); 
    346                 sequence.add(new Event<String>("b")); 
    347                 sequence.add(new Event<String>("r")); 
    348                 sequence.add(new Event<String>("a")); 
    349                 sequence.add(new Event<String>("c")); 
    350                 sequence.add(new Event<String>("a")); 
    351                 sequence.add(new Event<String>("d")); 
    352                 sequence.add(new Event<String>("a")); 
    353                 sequence.add(new Event<String>("b")); 
    354                 sequence.add(new Event<String>("r")); 
    355                 sequence.add(new Event<String>("a")); 
    356  
    357                 sequences = new ArrayList<List<? extends Event<?>>>(); 
     345                List<Event> sequence = new ArrayList<Event>(); 
     346                sequence.add(new Event(new StringEventType("a"))); 
     347                sequence.add(new Event(new StringEventType("b"))); 
     348                sequence.add(new Event(new StringEventType("r"))); 
     349                sequence.add(new Event(new StringEventType("a"))); 
     350                sequence.add(new Event(new StringEventType("c"))); 
     351                sequence.add(new Event(new StringEventType("a"))); 
     352                sequence.add(new Event(new StringEventType("d"))); 
     353                sequence.add(new Event(new StringEventType("a"))); 
     354                sequence.add(new Event(new StringEventType("b"))); 
     355                sequence.add(new Event(new StringEventType("r"))); 
     356                sequence.add(new Event(new StringEventType("a"))); 
     357 
     358                sequences = new ArrayList<List<Event>>(); 
    358359                sequences.add(sequence); 
    359360        } 

  • trunk/quest-core-usageprofiles-test/src/test/java/de/ugoe/cs/quest/usageprofiles/TrieBasedModelTest.java

    r518 r548  
    1111import de.ugoe.cs.quest.usageprofiles.TrieBasedModel; 
    1212import de.ugoe.cs.quest.usageprofiles.TrieNode; 
     13import de.ugoe.cs.quest.usageprofiles.mock.StringEventType; 
    1314 
    1415import org.junit.*; 
    … …  
    2425public class TrieBasedModelTest { 
    2526 
    26         List<Event<?>> sequence; 
    27         Collection<Event<?>> symbols; 
    28  
    29         private void assertTrieStructure(Trie<Event<?>> trie, int numSequences) { 
    30                 TrieNode<Event<?>> root = trie.find(null); 
    31                 TrieNode<Event<?>> root_a = root.getChild(new Event<String>("a")); 
    32                 TrieNode<Event<?>> root_a_a = root_a.getChild(new Event<String>("a")); 
    33                 TrieNode<Event<?>> root_a_b = root_a.getChild(new Event<String>("b")); 
    34                 TrieNode<Event<?>> root_a_b_a = root_a_b 
    35                                 .getChild(new Event<String>("a")); 
    36                 TrieNode<Event<?>> root_a_b_b = root_a_b 
    37                                 .getChild(new Event<String>("b")); 
    38                 TrieNode<Event<?>> root_a_b_c = root_a_b 
    39                                 .getChild(new Event<String>("c")); 
    40                 TrieNode<Event<?>> root_a_b_d = root_a_b 
    41                                 .getChild(new Event<String>("d")); 
    42                 TrieNode<Event<?>> root_a_b_r = root_a_b 
    43                                 .getChild(new Event<String>("r")); 
    44                 TrieNode<Event<?>> root_a_c = root_a.getChild(new Event<String>("c")); 
    45                 TrieNode<Event<?>> root_a_c_a = root_a_c 
    46                                 .getChild(new Event<String>("a")); 
    47                 TrieNode<Event<?>> root_a_c_b = root_a_c 
    48                                 .getChild(new Event<String>("b")); 
    49                 TrieNode<Event<?>> root_a_c_c = root_a_c 
    50                                 .getChild(new Event<String>("c")); 
    51                 TrieNode<Event<?>> root_a_c_d = root_a_c 
    52                                 .getChild(new Event<String>("d")); 
    53                 TrieNode<Event<?>> root_a_c_r = root_a_c 
    54                                 .getChild(new Event<String>("r")); 
    55                 TrieNode<Event<?>> root_a_d = root_a.getChild(new Event<String>("d")); 
    56                 TrieNode<Event<?>> root_a_d_a = root_a_d 
    57                                 .getChild(new Event<String>("a")); 
    58                 TrieNode<Event<?>> root_a_d_b = root_a_d 
    59                                 .getChild(new Event<String>("b")); 
    60                 TrieNode<Event<?>> root_a_d_c = root_a_d 
    61                                 .getChild(new Event<String>("c")); 
    62                 TrieNode<Event<?>> root_a_d_d = root_a_d 
    63                                 .getChild(new Event<String>("d")); 
    64                 TrieNode<Event<?>> root_a_d_r = root_a_d 
    65                                 .getChild(new Event<String>("r")); 
    66                 TrieNode<Event<?>> root_a_r = root_a.getChild(new Event<String>("r")); 
    67                 TrieNode<Event<?>> root_b = root.getChild(new Event<String>("b")); 
    68                 TrieNode<Event<?>> root_b_a = root_b.getChild(new Event<String>("a")); 
    69                 TrieNode<Event<?>> root_b_b = root_b.getChild(new Event<String>("b")); 
    70                 TrieNode<Event<?>> root_b_c = root_b.getChild(new Event<String>("c")); 
    71                 TrieNode<Event<?>> root_b_d = root_b.getChild(new Event<String>("d")); 
    72                 TrieNode<Event<?>> root_b_r = root_b.getChild(new Event<String>("r")); 
    73                 TrieNode<Event<?>> root_b_r_a = root_b_r 
    74                                 .getChild(new Event<String>("a")); 
    75                 TrieNode<Event<?>> root_b_r_b = root_b_r 
    76                                 .getChild(new Event<String>("b")); 
    77                 TrieNode<Event<?>> root_b_r_c = root_b_r 
    78                                 .getChild(new Event<String>("c")); 
    79                 TrieNode<Event<?>> root_b_r_d = root_b_r 
    80                                 .getChild(new Event<String>("d")); 
    81                 TrieNode<Event<?>> root_b_r_r = root_b_r 
    82                                 .getChild(new Event<String>("r")); 
    83                 TrieNode<Event<?>> root_c = root.getChild(new Event<String>("c")); 
    84                 TrieNode<Event<?>> root_c_a = root_c.getChild(new Event<String>("a")); 
    85                 TrieNode<Event<?>> root_c_a_a = root_c_a 
    86                                 .getChild(new Event<String>("a")); 
    87                 TrieNode<Event<?>> root_c_a_b = root_c_a 
    88                                 .getChild(new Event<String>("b")); 
    89                 TrieNode<Event<?>> root_c_a_c = root_c_a 
    90                                 .getChild(new Event<String>("c")); 
    91                 TrieNode<Event<?>> root_c_a_d = root_c_a 
    92                                 .getChild(new Event<String>("d")); 
    93                 TrieNode<Event<?>> root_c_a_r = root_c_a 
    94                                 .getChild(new Event<String>("r")); 
    95                 TrieNode<Event<?>> root_c_b = root_c.getChild(new Event<String>("b")); 
    96                 TrieNode<Event<?>> root_c_c = root_c.getChild(new Event<String>("c")); 
    97                 TrieNode<Event<?>> root_c_d = root_c.getChild(new Event<String>("d")); 
    98                 TrieNode<Event<?>> root_c_r = root_c.getChild(new Event<String>("r")); 
    99                 TrieNode<Event<?>> root_d = root.getChild(new Event<String>("d")); 
    100                 TrieNode<Event<?>> root_d_a = root_d.getChild(new Event<String>("a")); 
    101                 TrieNode<Event<?>> root_d_a_a = root_d_a 
    102                                 .getChild(new Event<String>("a")); 
    103                 TrieNode<Event<?>> root_d_a_b = root_d_a 
    104                                 .getChild(new Event<String>("b")); 
    105                 TrieNode<Event<?>> root_d_a_c = root_d_a 
    106                                 .getChild(new Event<String>("c")); 
    107                 TrieNode<Event<?>> root_d_a_d = root_d_a 
    108                                 .getChild(new Event<String>("d")); 
    109                 TrieNode<Event<?>> root_d_a_r = root_d_a 
    110                                 .getChild(new Event<String>("r")); 
    111                 TrieNode<Event<?>> root_d_b = root_d.getChild(new Event<String>("b")); 
    112                 TrieNode<Event<?>> root_d_c = root_d.getChild(new Event<String>("c")); 
    113                 TrieNode<Event<?>> root_d_d = root_d.getChild(new Event<String>("d")); 
    114                 TrieNode<Event<?>> root_d_r = root_d.getChild(new Event<String>("r")); 
    115                 TrieNode<Event<?>> root_r = root.getChild(new Event<String>("r")); 
    116                 TrieNode<Event<?>> root_r_a = root_r.getChild(new Event<String>("a")); 
    117                 TrieNode<Event<?>> root_r_a_a = root_r_a 
    118                                 .getChild(new Event<String>("a")); 
    119                 TrieNode<Event<?>> root_r_a_b = root_r_a 
    120                                 .getChild(new Event<String>("b")); 
    121                 TrieNode<Event<?>> root_r_a_c = root_r_a 
    122                                 .getChild(new Event<String>("c")); 
    123                 TrieNode<Event<?>> root_r_a_d = root_r_a 
    124                                 .getChild(new Event<String>("d")); 
    125                 TrieNode<Event<?>> root_r_a_r = root_r_a 
    126                                 .getChild(new Event<String>("r")); 
    127                 TrieNode<Event<?>> root_r_a_end = root_r_a.getChild(Event.ENDEVENT); 
    128                 TrieNode<Event<?>> root_r_b = root_r.getChild(new Event<String>("b")); 
    129                 TrieNode<Event<?>> root_r_c = root_r.getChild(new Event<String>("c")); 
    130                 TrieNode<Event<?>> root_r_d = root_r.getChild(new Event<String>("d")); 
    131                 TrieNode<Event<?>> root_r_r = root_r.getChild(new Event<String>("r")); 
    132                 TrieNode<Event<?>> root_start = root.getChild(Event.STARTEVENT); 
    133                 TrieNode<Event<?>> root_start_a = root_start 
    134                                 .getChild(new Event<String>("a")); 
    135                 TrieNode<Event<?>> root_start_a_a = root_start_a 
    136                                 .getChild(new Event<String>("a")); 
    137                 TrieNode<Event<?>> root_start_a_b = root_start_a 
    138                                 .getChild(new Event<String>("b")); 
    139                 TrieNode<Event<?>> root_start_a_c = root_start_a 
    140                                 .getChild(new Event<String>("c")); 
    141                 TrieNode<Event<?>> root_start_a_d = root_start_a 
    142                                 .getChild(new Event<String>("d")); 
    143                 TrieNode<Event<?>> root_start_a_r = root_start_a 
    144                                 .getChild(new Event<String>("r")); 
    145                 TrieNode<Event<?>> root_start_b = root_start 
    146                                 .getChild(new Event<String>("b")); 
    147                 TrieNode<Event<?>> root_start_c = root_start 
    148                                 .getChild(new Event<String>("c")); 
    149                 TrieNode<Event<?>> root_start_d = root_start 
    150                                 .getChild(new Event<String>("d")); 
    151                 TrieNode<Event<?>> root_start_r = root_start 
    152                                 .getChild(new Event<String>("r")); 
     27        List<Event> sequence; 
     28        Collection<Event> symbols; 
     29 
     30        private void assertTrieStructure(Trie<Event> trie, int numSequences) { 
     31                TrieNode<Event> root = trie.find(null); 
     32                TrieNode<Event> root_a = root.getChild(new Event(new StringEventType("a"))); 
     33                TrieNode<Event> root_a_a = root_a.getChild(new Event(new StringEventType("a"))); 
     34                TrieNode<Event> root_a_b = root_a.getChild(new Event(new StringEventType("b"))); 
     35                TrieNode<Event> root_a_b_a = root_a_b 
     36                                .getChild(new Event(new StringEventType("a"))); 
     37                TrieNode<Event> root_a_b_b = root_a_b 
     38                                .getChild(new Event(new StringEventType("b"))); 
     39                TrieNode<Event> root_a_b_c = root_a_b 
     40                                .getChild(new Event(new StringEventType("c"))); 
     41                TrieNode<Event> root_a_b_d = root_a_b 
     42                                .getChild(new Event(new StringEventType("d"))); 
     43                TrieNode<Event> root_a_b_r = root_a_b 
     44                                .getChild(new Event(new StringEventType("r"))); 
     45                TrieNode<Event> root_a_c = root_a.getChild(new Event(new StringEventType("c"))); 
     46                TrieNode<Event> root_a_c_a = root_a_c 
     47                                .getChild(new Event(new StringEventType("a"))); 
     48                TrieNode<Event> root_a_c_b = root_a_c 
     49                                .getChild(new Event(new StringEventType("b"))); 
     50                TrieNode<Event> root_a_c_c = root_a_c 
     51                                .getChild(new Event(new StringEventType("c"))); 
     52                TrieNode<Event> root_a_c_d = root_a_c 
     53                                .getChild(new Event(new StringEventType("d"))); 
     54                TrieNode<Event> root_a_c_r = root_a_c 
     55                                .getChild(new Event(new StringEventType("r"))); 
     56                TrieNode<Event> root_a_d = root_a.getChild(new Event(new StringEventType("d"))); 
     57                TrieNode<Event> root_a_d_a = root_a_d 
     58                                .getChild(new Event(new StringEventType("a"))); 
     59                TrieNode<Event> root_a_d_b = root_a_d 
     60                                .getChild(new Event(new StringEventType("b"))); 
     61                TrieNode<Event> root_a_d_c = root_a_d 
     62                                .getChild(new Event(new StringEventType("c"))); 
     63                TrieNode<Event> root_a_d_d = root_a_d 
     64                                .getChild(new Event(new StringEventType("d"))); 
     65                TrieNode<Event> root_a_d_r = root_a_d 
     66                                .getChild(new Event(new StringEventType("r"))); 
     67                TrieNode<Event> root_a_r = root_a.getChild(new Event(new StringEventType("r"))); 
     68                TrieNode<Event> root_b = root.getChild(new Event(new StringEventType("b"))); 
     69                TrieNode<Event> root_b_a = root_b.getChild(new Event(new StringEventType("a"))); 
     70                TrieNode<Event> root_b_b = root_b.getChild(new Event(new StringEventType("b"))); 
     71                TrieNode<Event> root_b_c = root_b.getChild(new Event(new StringEventType("c"))); 
     72                TrieNode<Event> root_b_d = root_b.getChild(new Event(new StringEventType("d"))); 
     73                TrieNode<Event> root_b_r = root_b.getChild(new Event(new StringEventType("r"))); 
     74                TrieNode<Event> root_b_r_a = root_b_r 
     75                                .getChild(new Event(new StringEventType("a"))); 
     76                TrieNode<Event> root_b_r_b = root_b_r 
     77                                .getChild(new Event(new StringEventType("b"))); 
     78                TrieNode<Event> root_b_r_c = root_b_r 
     79                                .getChild(new Event(new StringEventType("c"))); 
     80                TrieNode<Event> root_b_r_d = root_b_r 
     81                                .getChild(new Event(new StringEventType("d"))); 
     82                TrieNode<Event> root_b_r_r = root_b_r 
     83                                .getChild(new Event(new StringEventType("r"))); 
     84                TrieNode<Event> root_c = root.getChild(new Event(new StringEventType("c"))); 
     85                TrieNode<Event> root_c_a = root_c.getChild(new Event(new StringEventType("a"))); 
     86                TrieNode<Event> root_c_a_a = root_c_a 
     87                                .getChild(new Event(new StringEventType("a"))); 
     88                TrieNode<Event> root_c_a_b = root_c_a 
     89                                .getChild(new Event(new StringEventType("b"))); 
     90                TrieNode<Event> root_c_a_c = root_c_a 
     91                                .getChild(new Event(new StringEventType("c"))); 
     92                TrieNode<Event> root_c_a_d = root_c_a 
     93                                .getChild(new Event(new StringEventType("d"))); 
     94                TrieNode<Event> root_c_a_r = root_c_a 
     95                                .getChild(new Event(new StringEventType("r"))); 
     96                TrieNode<Event> root_c_b = root_c.getChild(new Event(new StringEventType("b"))); 
     97                TrieNode<Event> root_c_c = root_c.getChild(new Event(new StringEventType("c"))); 
     98                TrieNode<Event> root_c_d = root_c.getChild(new Event(new StringEventType("d"))); 
     99                TrieNode<Event> root_c_r = root_c.getChild(new Event(new StringEventType("r"))); 
     100                TrieNode<Event> root_d = root.getChild(new Event(new StringEventType("d"))); 
     101                TrieNode<Event> root_d_a = root_d.getChild(new Event(new StringEventType("a"))); 
     102                TrieNode<Event> root_d_a_a = root_d_a 
     103                                .getChild(new Event(new StringEventType("a"))); 
     104                TrieNode<Event> root_d_a_b = root_d_a 
     105                                .getChild(new Event(new StringEventType("b"))); 
     106                TrieNode<Event> root_d_a_c = root_d_a 
     107                                .getChild(new Event(new StringEventType("c"))); 
     108                TrieNode<Event> root_d_a_d = root_d_a 
     109                                .getChild(new Event(new StringEventType("d"))); 
     110                TrieNode<Event> root_d_a_r = root_d_a 
     111                                .getChild(new Event(new StringEventType("r"))); 
     112                TrieNode<Event> root_d_b = root_d.getChild(new Event(new StringEventType("b"))); 
     113                TrieNode<Event> root_d_c = root_d.getChild(new Event(new StringEventType("c"))); 
     114                TrieNode<Event> root_d_d = root_d.getChild(new Event(new StringEventType("d"))); 
     115                TrieNode<Event> root_d_r = root_d.getChild(new Event(new StringEventType("r"))); 
     116                TrieNode<Event> root_r = root.getChild(new Event(new StringEventType("r"))); 
     117                TrieNode<Event> root_r_a = root_r.getChild(new Event(new StringEventType("a"))); 
     118                TrieNode<Event> root_r_a_a = root_r_a 
     119                                .getChild(new Event(new StringEventType("a"))); 
     120                TrieNode<Event> root_r_a_b = root_r_a 
     121                                .getChild(new Event(new StringEventType("b"))); 
     122                TrieNode<Event> root_r_a_c = root_r_a 
     123                                .getChild(new Event(new StringEventType("c"))); 
     124                TrieNode<Event> root_r_a_d = root_r_a 
     125                                .getChild(new Event(new StringEventType("d"))); 
     126                TrieNode<Event> root_r_a_r = root_r_a 
     127                                .getChild(new Event(new StringEventType("r"))); 
     128                TrieNode<Event> root_r_a_end = root_r_a.getChild(Event.ENDEVENT); 
     129                TrieNode<Event> root_r_b = root_r.getChild(new Event(new StringEventType("b"))); 
     130                TrieNode<Event> root_r_c = root_r.getChild(new Event(new StringEventType("c"))); 
     131                TrieNode<Event> root_r_d = root_r.getChild(new Event(new StringEventType("d"))); 
     132                TrieNode<Event> root_r_r = root_r.getChild(new Event(new StringEventType("r"))); 
     133                TrieNode<Event> root_start = root.getChild(Event.STARTEVENT); 
     134                TrieNode<Event> root_start_a = root_start 
     135                                .getChild(new Event(new StringEventType("a"))); 
     136                TrieNode<Event> root_start_a_a = root_start_a 
     137                                .getChild(new Event(new StringEventType("a"))); 
     138                TrieNode<Event> root_start_a_b = root_start_a 
     139                                .getChild(new Event(new StringEventType("b"))); 
     140                TrieNode<Event> root_start_a_c = root_start_a 
     141                                .getChild(new Event(new StringEventType("c"))); 
     142                TrieNode<Event> root_start_a_d = root_start_a 
     143                                .getChild(new Event(new StringEventType("d"))); 
     144                TrieNode<Event> root_start_a_r = root_start_a 
     145                                .getChild(new Event(new StringEventType("r"))); 
     146                TrieNode<Event> root_start_b = root_start 
     147                                .getChild(new Event(new StringEventType("b"))); 
     148                TrieNode<Event> root_start_c = root_start 
     149                                .getChild(new Event(new StringEventType("c"))); 
     150                TrieNode<Event> root_start_d = root_start 
     151                                .getChild(new Event(new StringEventType("d"))); 
     152                TrieNode<Event> root_start_r = root_start 
     153                                .getChild(new Event(new StringEventType("r"))); 
    153154 
    154155                assertEquals(numSequences * 5, root_a.getCount()); 
    … …  
    287288                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    288289                                new Random()); 
    289                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     290                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    290291                sequences.add(sequence); 
    291292                fixture.train(sequences); 
    292293                int length = 2; 
    293294 
    294                 Collection<List<Event<?>>> expected = new HashSet<List<Event<?>>>(); 
    295                 ArrayList<Event<?>> list; 
    296                 list = new ArrayList<Event<?>>(); 
    297                 list.add(new Event<String>("a")); 
     295                Collection<List<Event>> expected = new HashSet<List<Event>>(); 
     296                ArrayList<Event> list; 
     297                list = new ArrayList<Event>(); 
     298                list.add(new Event(new StringEventType("a"))); 
    298299                list.add(Event.ENDEVENT); 
    299300                expected.add(list); 
    300                 list = new ArrayList<Event<?>>(); 
    301                 list.add(new Event<String>("a")); 
    302                 list.add(new Event<String>("b")); 
    303                 expected.add(list); 
    304                 list = new ArrayList<Event<?>>(); 
    305                 list.add(new Event<String>("a")); 
    306                 list.add(new Event<String>("c")); 
    307                 expected.add(list); 
    308                 list = new ArrayList<Event<?>>(); 
    309                 list.add(new Event<String>("a")); 
    310                 list.add(new Event<String>("d")); 
    311                 expected.add(list); 
    312                 list = new ArrayList<Event<?>>(); 
    313                 list.add(new Event<String>("b")); 
    314                 list.add(new Event<String>("r")); 
    315                 expected.add(list); 
    316                 list = new ArrayList<Event<?>>(); 
    317                 list.add(new Event<String>("c")); 
    318                 list.add(new Event<String>("a")); 
    319                 expected.add(list); 
    320                 list = new ArrayList<Event<?>>(); 
    321                 list.add(new Event<String>("d")); 
    322                 list.add(new Event<String>("a")); 
    323                 expected.add(list); 
    324                 list = new ArrayList<Event<?>>(); 
    325                 list.add(new Event<String>("r")); 
    326                 list.add(new Event<String>("a")); 
    327                 expected.add(list); 
    328                 list = new ArrayList<Event<?>>(); 
     301                list = new ArrayList<Event>(); 
     302                list.add(new Event(new StringEventType("a"))); 
     303                list.add(new Event(new StringEventType("b"))); 
     304                expected.add(list); 
     305                list = new ArrayList<Event>(); 
     306                list.add(new Event(new StringEventType("a"))); 
     307                list.add(new Event(new StringEventType("c"))); 
     308                expected.add(list); 
     309                list = new ArrayList<Event>(); 
     310                list.add(new Event(new StringEventType("a"))); 
     311                list.add(new Event(new StringEventType("d"))); 
     312                expected.add(list); 
     313                list = new ArrayList<Event>(); 
     314                list.add(new Event(new StringEventType("b"))); 
     315                list.add(new Event(new StringEventType("r"))); 
     316                expected.add(list); 
     317                list = new ArrayList<Event>(); 
     318                list.add(new Event(new StringEventType("c"))); 
     319                list.add(new Event(new StringEventType("a"))); 
     320                expected.add(list); 
     321                list = new ArrayList<Event>(); 
     322                list.add(new Event(new StringEventType("d"))); 
     323                list.add(new Event(new StringEventType("a"))); 
     324                expected.add(list); 
     325                list = new ArrayList<Event>(); 
     326                list.add(new Event(new StringEventType("r"))); 
     327                list.add(new Event(new StringEventType("a"))); 
     328                expected.add(list); 
     329                list = new ArrayList<Event>(); 
    329330                list.add(Event.STARTEVENT); 
    330                 list.add(new Event<String>("a")); 
    331                 expected.add(list); 
    332  
    333                 Collection<List<? extends Event<?>>> result = fixture 
     331                list.add(new Event(new StringEventType("a"))); 
     332                expected.add(list); 
     333 
     334                Collection<List<Event>> result = fixture 
    334335                                .generateSequences(length); 
    335336 
    … …  
    342343                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    343344                                new Random()); 
    344                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     345                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    345346                sequences.add(sequence); 
    346347                fixture.train(sequences); 
    347348                int length = 3; 
    348349 
    349                 Collection<List<Event<?>>> expected = new HashSet<List<Event<?>>>(); 
    350                 ArrayList<Event<?>> list; 
    351                 list = new ArrayList<Event<?>>(); 
     350                Collection<List<Event>> expected = new HashSet<List<Event>>(); 
     351                ArrayList<Event> list; 
     352                list = new ArrayList<Event>(); 
    352353                list.add(Event.STARTEVENT); 
    353                 list.add(new Event<String>("a")); 
     354                list.add(new Event(new StringEventType("a"))); 
    354355                list.add(Event.ENDEVENT); 
    355356                expected.add(list); 
    356                 list = new ArrayList<Event<?>>(); 
     357                list = new ArrayList<Event>(); 
    357358                list.add(Event.STARTEVENT); 
    358                 list.add(new Event<String>("a")); 
    359                 list.add(new Event<String>("b")); 
    360                 expected.add(list); 
    361                 list = new ArrayList<Event<?>>(); 
     359                list.add(new Event(new StringEventType("a"))); 
     360                list.add(new Event(new StringEventType("b"))); 
     361                expected.add(list); 
     362                list = new ArrayList<Event>(); 
    362363                list.add(Event.STARTEVENT); 
    363                 list.add(new Event<String>("a")); 
    364                 list.add(new Event<String>("c")); 
    365                 expected.add(list); 
    366                 list = new ArrayList<Event<?>>(); 
     364                list.add(new Event(new StringEventType("a"))); 
     365                list.add(new Event(new StringEventType("c"))); 
     366                expected.add(list); 
     367                list = new ArrayList<Event>(); 
    367368                list.add(Event.STARTEVENT); 
    368                 list.add(new Event<String>("a")); 
    369                 list.add(new Event<String>("d")); 
    370                 expected.add(list); 
    371  
    372                 Collection<List<? extends Event<?>>> result = fixture 
     369                list.add(new Event(new StringEventType("a"))); 
     370                list.add(new Event(new StringEventType("d"))); 
     371                expected.add(list); 
     372 
     373                Collection<List<Event>> result = fixture 
    373374                                .generateSequences(length, true); 
    374375 
    … …  
    381382                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    382383                                new Random()); 
    383                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     384                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    384385                sequences.add(sequence); 
    385386                fixture.train(sequences); 
    … …  
    394395                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    395396                                new Random()); 
    396                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     397                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    397398                sequences.add(sequence); 
    398399                fixture.train(sequences); 
    399400                int length = 5; 
    400401 
    401                 Collection<List<Event<?>>> expected = new HashSet<List<Event<?>>>(); 
    402                 ArrayList<Event<?>> list; 
    403                 list = new ArrayList<Event<?>>(); 
     402                Collection<List<Event>> expected = new HashSet<List<Event>>(); 
     403                ArrayList<Event> list; 
     404                list = new ArrayList<Event>(); 
    404405                list.add(Event.STARTEVENT); 
    405                 list.add(new Event<String>("a")); 
    406                 list.add(new Event<String>("c")); 
    407                 list.add(new Event<String>("a")); 
     406                list.add(new Event(new StringEventType("a"))); 
     407                list.add(new Event(new StringEventType("c"))); 
     408                list.add(new Event(new StringEventType("a"))); 
    408409                list.add(Event.ENDEVENT); 
    409410                expected.add(list); 
    410                 list = new ArrayList<Event<?>>(); 
     411                list = new ArrayList<Event>(); 
    411412                list.add(Event.STARTEVENT); 
    412                 list.add(new Event<String>("a")); 
    413                 list.add(new Event<String>("d")); 
    414                 list.add(new Event<String>("a")); 
     413                list.add(new Event(new StringEventType("a"))); 
     414                list.add(new Event(new StringEventType("d"))); 
     415                list.add(new Event(new StringEventType("a"))); 
    415416                list.add(Event.ENDEVENT); 
    416417                expected.add(list); 
    417418 
    418                 Collection<List<? extends Event<?>>> result = fixture 
     419                Collection<List<Event>> result = fixture 
    419420                                .generateValidSequences(length); 
    420421 
    … …  
    427428                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    428429                                new Random()); 
    429                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     430                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    430431                sequences.add(sequence); 
    431432                fixture.train(sequences); 
    … …  
    440441                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    441442                                new Random()); 
    442                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
    443                 sequences.add(sequence); 
    444  
    445                 fixture.train(sequences); 
    446  
    447                 Collection<? extends Event<?>> result = fixture.getEvents(); 
     443                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
     444                sequences.add(sequence); 
     445 
     446                fixture.train(sequences); 
     447 
     448                Collection<Event> result = fixture.getEvents(); 
    448449 
    449450                assertCollectionContent(symbols, result); 
    … …  
    456457                                new Random()); 
    457458 
    458                 Collection<? extends Event<?>> result = fixture.getEvents(); 
    459  
    460                 assertCollectionContent(new HashSet<Event<?>>(), result); 
     459                Collection<Event> result = fixture.getEvents(); 
     460 
     461                assertCollectionContent(new HashSet<Event>(), result); 
    461462        } 
    462463 
    … …  
    466467                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    467468                                new Random()); 
    468                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     469                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    469470                sequences.add(sequence); 
    470471 
    … …  
    493494                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    494495                                new Random()); 
    495                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     496                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    496497                sequences.add(sequence); 
    497498                fixture.train(sequences); 
    … …  
    518519                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    519520                                new Random()); 
    520                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     521                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    521522                sequences.add(sequence); 
    522523                fixture.train(sequences); 
    … …  
    543544                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    544545                                new Random()); 
    545                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     546                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    546547                sequences.add(sequence); 
    547548 
    … …  
    558559                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    559560                                new Random()); 
    560                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     561                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    561562                sequences.add(sequence); 
    562563                sequences.add(sequence); 
    … …  
    574575                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    575576                                new Random()); 
    576                 Collection<List<? extends Event<?>>> sequences = null; 
     577                Collection<List<Event>> sequences = null; 
    577578 
    578579                fixture.train(sequences); 
    … …  
    584585                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    585586                                new Random()); 
    586                 Collection<List<? extends Event<?>>> sequences = new ArrayList<List<? extends Event<?>>>(); 
     587                Collection<List<Event>> sequences = new ArrayList<List<Event>>(); 
    587588                sequences.add(sequence); 
    588589                fixture.train(sequences); 
    … …  
    599600                MockTrieBasedModel fixture = new MockTrieBasedModel(markovOrder, 
    600601                                new Random()); 
    601                 Collection<List<? extends Event<?>>> sequences = null; 
     602                Collection<List<Event>> sequences = null; 
    602603                fixture.trie = null; 
    603604 
    … …  
    607608        @Before 
    608609        public void setUp() throws Exception { 
    609                 sequence = new ArrayList<Event<?>>(); 
    610                 sequence.add(new Event<String>("a")); 
    611                 sequence.add(new Event<String>("b")); 
    612                 sequence.add(new Event<String>("r")); 
    613                 sequence.add(new Event<String>("a")); 
    614                 sequence.add(new Event<String>("c")); 
    615                 sequence.add(new Event<String>("a")); 
    616                 sequence.add(new Event<String>("d")); 
    617                 sequence.add(new Event<String>("a")); 
    618                 sequence.add(new Event<String>("b")); 
    619                 sequence.add(new Event<String>("r")); 
    620                 sequence.add(new Event<String>("a")); 
    621  
    622                 symbols = new HashSet<Event<?>>(); 
    623                 symbols.add(new Event<String>("a")); 
    624                 symbols.add(new Event<String>("b")); 
    625                 symbols.add(new Event<String>("c")); 
    626                 symbols.add(new Event<String>("d")); 
    627                 symbols.add(new Event<String>("r")); 
     610                sequence = new ArrayList<Event>(); 
     611                sequence.add(new Event(new StringEventType("a"))); 
     612                sequence.add(new Event(new StringEventType("b"))); 
     613                sequence.add(new Event(new StringEventType("r"))); 
     614                sequence.add(new Event(new StringEventType("a"))); 
     615                sequence.add(new Event(new StringEventType("c"))); 
     616                sequence.add(new Event(new StringEventType("a"))); 
     617                sequence.add(new Event(new StringEventType("d"))); 
     618                sequence.add(new Event(new StringEventType("a"))); 
     619                sequence.add(new Event(new StringEventType("b"))); 
     620                sequence.add(new Event(new StringEventType("r"))); 
     621                sequence.add(new Event(new StringEventType("a"))); 
     622 
     623                symbols = new HashSet<Event>(); 
     624                symbols.add(new Event(new StringEventType("a"))); 
     625                symbols.add(new Event(new StringEventType("b"))); 
     626                symbols.add(new Event(new StringEventType("c"))); 
     627                symbols.add(new Event(new StringEventType("d"))); 
     628                symbols.add(new Event(new StringEventType("r"))); 
    628629                symbols.add(Event.STARTEVENT); 
    629630                symbols.add(Event.ENDEVENT); 
Note: See TracChangeset for help on using the changeset viewer.