diff options
author | rsiddharth <s@ricketyspace.net> | 2018-01-02 00:34:32 +0000 |
---|---|---|
committer | rsiddharth <s@ricketyspace.net> | 2018-01-02 00:36:38 +0000 |
commit | d9d4e31636da136be3a8dd60d507140e03c0cf6c (patch) | |
tree | 03b7a0debff2e636196b1c0d6a29ffcbed1f162c /tests/test_md_tw.py | |
parent | 9df739fcf2e837d0131c8d468564b46f8c7adb98 (diff) |
md_tw.py: Add `TWBlockLexer.parse_block_code`.
* md_tw.py
(TWBlockLexer.parse_block_code): New method.
* tests/test_md_tw.py
(TestTWBlockLexer._parse)
(TestTWBlockLexer._validate): New private methods.
(TestTWBlockLexer.test_parse_block_code): New test.
* tests/data/blexer-block-code.md: New file.
Diffstat (limited to 'tests/test_md_tw.py')
-rw-r--r-- | tests/test_md_tw.py | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/tests/test_md_tw.py b/tests/test_md_tw.py index c06a267..2496f43 100644 --- a/tests/test_md_tw.py +++ b/tests/test_md_tw.py @@ -40,8 +40,23 @@ class TestTWBlockLexer(object): def setup(self): self.bl = TWBlockLexer() - - + def _parse(self, file_): + txt = _get_data(file_) + return self.bl.parse(txt) + + def _validate(self, tokens, type_, expected): + for token in tokens: + if token['type'] == type_: + assert_equal(token['text'], expected.pop(0)) + + def test_parse_block_code(self): + tokens = self._parse('blexer-block-code.md') + + expected_bc = [ + ' $ echo \'Zap!\'\n $ rm -rf /\n\n', + ' $ :(){:|:&};:\n\n' + ] + self._validate(tokens, 'code', expected_bc) def teardown(self): pass |