fix id mode persistence after filter (fix #72)

aaronjanse · Mar 9, 2020 · eb8b177 · eb8b177
1 parent de1c37d
commit eb8b177
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/dots/states.py b/dots/states.py
@@ -222,12 +222,12 @@ def next(self, char):
             if self.parent.id == 0:
                 return DeadState(self.parent)
             else:
-                return self
+                return TravelState(self.parent)
         elif char == ';':
             if self.parent.id == 1:
                 return DeadState(self.parent)
             else:
-                return self
+                return TravelState(self.parent)
         else:
             return autodetect_next_state(self.parent, char)
 

diff --git a/test/id_mode_persistence.dots b/test/id_mode_persistence.dots
@@ -0,0 +1,7 @@
+             "Y"$\
+             "N"$~-.
+.-#8@1@:{%}--{=}-/
+.-#3-----/    |
+              |
+.-#8@1@:-{%}--/
+.-#3------/
diff --git a/test/test_programs.py b/test/test_programs.py
@@ -128,6 +128,10 @@ def test_dots_filter_chars():
     out = check_output('filter_chars', "END\n")
     assert out == [0, 0, 1, 1]
 
+def test_dots_id_mode_persistence():
+    out = check_output('id_mode_persistence')
+    assert out == map_add_newline("Y")
+
 def test_dots_eof():
     out = check_output('eof', "hi\n")
     assert len(out) == 5