Added cases to Transition.getEntrySet, Transition.getExitSet.

clnhlzmn · Mar 24, 2021 · ff41b2e · ff41b2e
1 parent 552ab70
commit ff41b2e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/makina-compiler/test/xyz/colinholzman/makina/TransitionTest.kt b/makina-compiler/test/xyz/colinholzman/makina/TransitionTest.kt
@@ -19,6 +19,8 @@ internal class TransitionTest {
         assertEquals(listOf(s12, s122), transition.getEntrySet())
         transition = Transition(s111, s21)
         assertEquals(listOf(s2, s21), transition.getEntrySet())
+        transition = Transition(s111, s111)
+        assertEquals(listOf(s111), transition.getEntrySet())
     }
 
     @Test
@@ -27,5 +29,7 @@ internal class TransitionTest {
         assertEquals(listOf(s111, s11), transition.getExitSet())
         transition = Transition(s111, s21)
         assertEquals(listOf(s111, s11, s1), transition.getExitSet())
+        transition = Transition(s111, s111)
+        assertEquals(listOf(s111), transition.getExitSet())
     }
 }