def f(w): collector = Collector(w.spec_root_path) imported_files = [] def mock_import_file_func(path, root): imported_files.append(path) collector.import_spec_file = mock_import_file_func collector.import_spec_files() expected = ['a_spec.py', 'b/b_spec.py', 'c/d/d_spec.py'] expected = set(['%s/%s' % (w.spec_root_path, x) for x in expected]) assert set(imported_files) == expected, '%r != %r' % (imported_files, expected)