1818from bpython import config
1919from bpython import args
2020from bpython ._py3compat import py3
21+ from bpython .test import FixLanguageTestCase as TestCase
2122
2223def setup_config (conf ):
2324 config_struct = config .Struct ()
@@ -29,7 +30,7 @@ def setup_config(conf):
2930 return config_struct
3031
3132
32- class TestCurtsiesRepl (unittest . TestCase ):
33+ class TestCurtsiesRepl (TestCase ):
3334
3435 def setUp (self ):
3536 self .repl = create_repl ()
@@ -88,7 +89,7 @@ def mock_next(obj, return_value):
8889 else :
8990 obj .next .return_value = return_value
9091
91- class TestCurtsiesReplTab (unittest . TestCase ):
92+ class TestCurtsiesReplTab (TestCase ):
9293
9394 def setUp (self ):
9495 self .repl = create_repl ()
@@ -153,7 +154,7 @@ def test_tab_completes_common_sequence(self):
153154 self .repl .matches_iter .substitute_cseq .assert_called_once_with ()
154155
155156
156- class TestCurtsiesReplFilenameCompletion (unittest . TestCase ):
157+ class TestCurtsiesReplFilenameCompletion (TestCase ):
157158 def setUp (self ):
158159 self .repl = create_repl ()
159160
@@ -213,7 +214,7 @@ def create_repl(**kwargs):
213214 repl .height = 20
214215 return repl
215216
216- class TestFutureImports (unittest . TestCase ):
217+ class TestFutureImports (TestCase ):
217218
218219 def test_repl (self ):
219220 repl = create_repl ()
@@ -236,7 +237,7 @@ def test_interactive(self):
236237
237238 self .assertEqual (out .getvalue (), '0.5\n 0.5\n ' )
238239
239- class TestPredictedIndent (unittest . TestCase ):
240+ class TestPredictedIndent (TestCase ):
240241 def setUp (self ):
241242 self .repl = create_repl ()
242243
@@ -253,7 +254,7 @@ def test_complex(self):
253254 self .assertEqual (self .repl .predicted_indent ('reduce(asdfasdf,' ), 7 )
254255
255256
256- class TestCurtsiesReevaluate (unittest . TestCase ):
257+ class TestCurtsiesReevaluate (TestCase ):
257258 def setUp (self ):
258259 self .repl = create_repl ()
259260
@@ -264,7 +265,7 @@ def test_variable_is_cleared(self):
264265 self .repl .undo ()
265266 self .assertNotIn ('b' , self .repl .interp .locals )
266267
267- class TestCurtsiesPagerText (unittest . TestCase ):
268+ class TestCurtsiesPagerText (TestCase ):
268269
269270 def setUp (self ):
270271 self .repl = create_repl ()
0 commit comments