diff --git a/turinglab/input.py b/turinglab/input.py index 2ace6b9..3ad1693 100644 --- a/turinglab/input.py +++ b/turinglab/input.py @@ -20,8 +20,8 @@ def from_file(filename: str): if not action: continue - new_symbol, direction, new_state = re.split('([<|>|.])', action) - new_state = int(new_state) - 1 + new_symbol, direction, *new_state = action + new_state = int(''.join(new_state)) - 1 movement = { '>': Movemement.R, @@ -31,8 +31,6 @@ def from_file(filename: str): program.set(i, symbol, Action(new_state, new_symbol, movement)) - # program[symbol][i] = ([new_symbol, direction, new_state]) - f.close() return program