diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_cmdline.py | 4 | ||||
-rw-r--r-- | tests/test_token.py | 51 |
2 files changed, 55 insertions, 0 deletions
diff --git a/tests/test_cmdline.py b/tests/test_cmdline.py index f524e976..00572d96 100644 --- a/tests/test_cmdline.py +++ b/tests/test_cmdline.py @@ -62,6 +62,10 @@ class CmdLineTest(unittest.TestCase): self.assertEquals(c, 0) self.assert_('HTML' in o) + def test_S_opt(self): + c, o, e = run_cmdline("-S", "default", "-f", "html", "-O", "linenos=1") + self.assertEquals(c, 0) + def test_invalid_opts(self): for opts in [("-L", "-lpy"), ("-L", "-fhtml"), ("-L", "-Ox"), ("-a",), ("-Sst", "-lpy"), ("-H",), diff --git a/tests/test_token.py b/tests/test_token.py new file mode 100644 index 00000000..ac57126c --- /dev/null +++ b/tests/test_token.py @@ -0,0 +1,51 @@ +# -*- coding: utf-8 -*- +""" + Test suite for the token module + ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + + :copyright: 2006-2007 by Georg Brandl. + :license: BSD, see LICENSE for more details. +""" + +import unittest +import StringIO +import sys + +from pygments import token + + +class TokenTest(unittest.TestCase): + + def test_tokentype(self): + e = self.assertEquals + r = self.assertRaises + + t = token.String + + e(t.split(), [token.Token, token.Literal, token.String]) + + e(t.__class__, token._TokenType) + + def test_functions(self): + self.assert_(token.is_token_subtype(token.String, token.String)) + self.assert_(token.is_token_subtype(token.String, token.Literal)) + self.failIf(token.is_token_subtype(token.Literal, token.String)) + + self.assert_(token.string_to_tokentype(token.String) is token.String) + self.assert_(token.string_to_tokentype('') is token.Token) + self.assert_(token.string_to_tokentype('String') is token.String) + + def test_sanity_check(self): + try: + try: + old_stdout = sys.stdout + sys.stdout = StringIO.StringIO() + execfile(token.__file__.rstrip('c'), {'__name__': '__main__'}) + finally: + sys.stdout = old_stdout + except SystemExit: + pass + + +if __name__ == '__main__': + unittest.main() |