class TestLessLexer(unittest.TestCase): """ Unit tests for LessLexer """ def setUp(self): self.lexer = LessLexer() def test_input_stream(self): """ It can load content from a string. """ file = StringIO(""" @simple-var: 1; """) self.lexer.input(file) token = self.lexer.token() self.assertEqual('@simple-var', token.value) def test_input_path(self): """ It can load content from a path. """ file = NamedTemporaryFile() file.write(b""" @simple-var: 1; """) file.seek(0) self.lexer.input(file.name) token = self.lexer.token() self.assertEqual('@simple-var', token.value)
def setUp(self): self.lexer = LessLexer()