From d9d4e31636da136be3a8dd60d507140e03c0cf6c Mon Sep 17 00:00:00 2001 From: rsiddharth Date: Tue, 2 Jan 2018 00:34:32 +0000 Subject: 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. --- tests/test_md_tw.py | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) (limited to 'tests/test_md_tw.py') 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 -- cgit v1.2.3