From ff41b2ef4c570ff7cfbf2d52f305b77985cdc1d4 Mon Sep 17 00:00:00 2001
From: Colin <colin.holzman@protonmail.com>
Date: Tue, 23 Mar 2021 21:12:01 -0400
Subject: [PATCH] Added cases to Transition.getEntrySet, Transition.getExitSet.

---
 .../test/xyz/colinholzman/makina/TransitionTest.kt            | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/makina-compiler/test/xyz/colinholzman/makina/TransitionTest.kt b/makina-compiler/test/xyz/colinholzman/makina/TransitionTest.kt
index 947a97f..0aecb07 100644
--- 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())
     }
 }
\ No newline at end of file