Merged in protz/pygments-main/add-envname (pull request #235)

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..3aa50420
--- /dev/null
+++ b/tests/string_asserts.py
@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+"""
+    Pygments string assert utility
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    :copyright: Copyright 2006-2014 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))