def test_failover_in_case_of_exception(): class Iterator(object): def __iter__(self): raise FileNotFoundError() file = Substitute() file.yields(Iterator()) hosts = Hostnames(config=file, default=['a', 'b']) assert_equal(list(hosts._hosts()), ['a', 'b'])
def test_yields(): '''Substitute: How to provide data for iteration''' # System under Test substitute = Substitute() # Act: Configure substitute.yields(['a', 'b', 'c']) # Act: Run result = list(substitute) # Assert assert_equal(result, ['a', 'b', 'c'])
def test_ignore_tokens(): lexer = Substitute() lexer.yields([Number(3.1415), NewLine(), Comment('Test'), Number(3.1415)]) expected_tokens = [ Number(3.1415), Number(3.1415), ] # System under Test tokens = IgnoreTokens(lexer, tokens=[Comment, NewLine]) # Assert tokens = list(tokens) assert_equal(tokens, expected_tokens)