diff options
Diffstat (limited to 'test/test_pygen.py')
-rw-r--r-- | test/test_pygen.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/test/test_pygen.py b/test/test_pygen.py index 5200e3e..8adc142 100644 --- a/test/test_pygen.py +++ b/test/test_pygen.py @@ -1,12 +1,11 @@ from io import StringIO -import unittest from mako.pygen import adjust_whitespace from mako.pygen import PythonPrinter -from .util.assertions import eq_ +from mako.testing.assertions import eq_ -class GeneratePythonTest(unittest.TestCase): +class GeneratePythonTest: def test_generate_normal(self): stream = StringIO() printer = PythonPrinter(stream) @@ -165,7 +164,7 @@ print "more indent" ) -class WhitespaceTest(unittest.TestCase): +class WhitespaceTest: def test_basic(self): text = """ for x in range(0,15): |