diff options
Diffstat (limited to 'yarnlib')
-rw-r--r-- | yarnlib/block_parser.py | 5 | ||||
-rw-r--r-- | yarnlib/block_parser_tests.py | 4 |
2 files changed, 9 insertions, 0 deletions
diff --git a/yarnlib/block_parser.py b/yarnlib/block_parser.py index db99b31..df95fb2 100644 --- a/yarnlib/block_parser.py +++ b/yarnlib/block_parser.py @@ -44,6 +44,7 @@ class BlockParser(object): 'AND': self.parse_and, '...': self.parse_continuation, 'IMPLEMENTS': self.parse_implementing, + 'EXAMPLE': self.parse_example, } def parse_blocks(self, blocks): @@ -148,3 +149,7 @@ class BlockParser(object): self.implementations.append(implementation) return blocks + def parse_example(self, line, blocks): + if blocks: + del blocks[0] + return blocks diff --git a/yarnlib/block_parser_tests.py b/yarnlib/block_parser_tests.py index 689a197..ead8a31 100644 --- a/yarnlib/block_parser_tests.py +++ b/yarnlib/block_parser_tests.py @@ -30,6 +30,10 @@ class BlockParserTests(unittest.TestCase): self.assertEqual(self.parser.scenarios, []) self.assertEqual(self.parser.implementations, []) + def test_skips_examples(self): + self.parser.parse_blocks(['EXAMPLE foo', 'bar']) + self.assertEqual(len(self.parser.scenarios), 0) + def test_parses_simple_elements(self): self.parser.parse_blocks( ['SCENARIO foo', 'ASSUMING something', 'GIVEN bar', |