Merge pull request #14 from patxoca/issue-11

Fix issue 11

1 files changed, 23 insertions, 0 deletions

diff --git a/tests/test_z80count.py b/tests/test_z80count.py
new file mode 100644
index 0000000..85a9084
--- /dev/null
+++ b/tests/test_z80count.py
@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+
+import pytest
+
+from z80count.z80count import Parser
+from z80count.z80count import z80count
+
+
+@pytest.mark.parametrize("line,expected", (
+    ("PLY_InterruptionOn: call PLY_Init",
+     "PLY_InterruptionOn: call PLY_Init ; [17]\n"),
+    ("$PLY_Interruption.On: call PLY_Init",
+     "$PLY_Interruption.On: call PLY_Init ; [17]\n"),
+    ("PLY_ReplayFrequency:\tld de,0",
+     "PLY_ReplayFrequency:\tld de,0 ; [10]\n"),
+
+))
+def test_issue_11(line, expected):
+    parser = Parser()
+    output, _ = z80count(line, parser, total=0, subt=False,
+                         no_update=True, column=1, use_tabs=False,
+                         tab_width=4, debug=False)
+    assert output == expected