diff options
Diffstat (limited to 'tests/string_asserts.py')
| -rw-r--r-- | tests/string_asserts.py | 22 | 
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/string_asserts.py b/tests/string_asserts.py new file mode 100644 index 00000000..11f5c7f0 --- /dev/null +++ b/tests/string_asserts.py @@ -0,0 +1,22 @@ +# -*- coding: utf-8 -*- +""" +    Pygments string assert utility +    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +    :copyright: Copyright 2006-2015 by the Pygments team, see AUTHORS. +    :license: BSD, see LICENSE for details. +""" + +class StringTests(object): + +    def assertStartsWith(self, haystack, needle, msg=None): +        if msg is None: +            msg = "'{0}' does not start with '{1}'".format(haystack, needle) +        if not haystack.startswith(needle): +            raise(AssertionError(msg)) + +    def assertEndsWith(self, haystack, needle, msg=None): +        if msg is None: +            msg = "'{0}' does not end with '{1}'".format(haystack, needle) +        if not haystack.endswith(needle): +            raise(AssertionError(msg))  | 
