Added some extra tests

2 files changed, 10 insertions, 5 deletions

diff --git a/src/main.rs b/src/main.rs
index b6033c1..0fa03d5 100755
--- a/src/main.rs
+++ b/src/main.rs
@@ -10,7 +10,5 @@ fn test(regex: &str, input: &str) -> bool {
 }
 
 fn main() {
-    println!("{}", dbg!(test("(a|b)a", "aa")));
-    println!("{}", dbg!(test("(a|b)a", "ba")));
-    println!("{}", dbg!(test("(a|b)a", "bb")));
+    dbg!(test("regexisawesome", "regexisawesome"));
 }
diff --git a/src/nfa.rs b/src/nfa.rs
index 4757b83..eb06b88 100644
--- a/src/nfa.rs
+++ b/src/nfa.rs
@@ -121,7 +121,6 @@ impl NFA {
     }
 
     fn matches_helper(&self, state: &State, mut chars: Chars) -> bool {
-        println!("{state:?}\n{chars:?}");
         if self.accepting_states.contains(&state.id) {
             return chars.next().is_none();
         }
@@ -208,6 +207,14 @@ mod tests {
     }
 
     #[test]
+    fn test_star_empty_input() {
+        assert!(test("", ""));
+        assert!(test("a*", ""));
+        assert!(test("b*", ""));
+        assert!(test("(a|b)*", ""));
+    }
+
+    #[test]
     fn test_star_succ() {
         assert!(test("(0)*1(0)*", "000000000100000"));
         assert!(test("(a)*abc(a)*", "aaaaaaabcaaaaaa"));
@@ -224,7 +231,7 @@ mod tests {
 
     #[test]
     fn test_complex_succ() {
+        assert!(test("(a|b|c)*", "abababababcbcba"));
         assert!(test("(a|b)*cc", "aaabababaabacc"));
-        assert!(test("(0)*0101(1)*", "00000101111"));
     }
 }