diff --git a/tests/test_shells/pdb.module.ok b/tests/test_shells/pdb.module.ok index 1001dc42..7554dd6c 100644 --- a/tests/test_shells/pdb.module.ok +++ b/tests/test_shells/pdb.module.ok @@ -1,5 +1,3 @@ - 1  pdb-script.py:2   s --> class Foo(object):  1  pdb-script.py:6    --Call-- -> class Foo(object): diff --git a/tests/test_shells/pdb.subclass.ok b/tests/test_shells/pdb.subclass.ok index b6f024fd..d8eba5ef 100644 --- a/tests/test_shells/pdb.subclass.ok +++ b/tests/test_shells/pdb.subclass.ok @@ -1,9 +1,3 @@ ---Call-- --> def nop(_): - 2  :1   pdb-script.py:2    --> def nop(_): - 2  :1   pdb-script.py:2    --> class Foo(object):  2  :1   pdb-script.py:6    --Call-- -> class Foo(object): diff --git a/tests/test_shells/postproc.py b/tests/test_shells/postproc.py index 32f311d3..d544bf83 100755 --- a/tests/test_shells/postproc.py +++ b/tests/test_shells/postproc.py @@ -38,10 +38,11 @@ with codecs.open(fname, 'r', encoding='utf-8') as R: for line in (R if shell != 'fish' else R.read().split('\n')): i += 1 if not found_cd: - if shell == 'pdb': - found_cd = (i > 1) - else: - found_cd = ('cd tests/shell/3rd' in line) + found_cd = ( + 'class Foo(object):' in line + if shell == 'pdb' else + 'cd tests/shell/3rd' in line + ) continue if 'true is the last line' in line: break