Fix tests

This commit is contained in:
Dave Halter
2019-05-28 01:59:32 +02:00
parent 5fa8338886
commit 946869ab23
2 changed files with 3 additions and 3 deletions

View File

@@ -205,7 +205,7 @@ def infer_return_types(function_execution_context):
return ContextSet.from_sets(
ann.define_generics(type_var_dict)
if isinstance(ann, (AbstractAnnotatedClass, TypeVar)) else ann
if isinstance(ann, (AbstractAnnotatedClass, TypeVar)) else ContextSet({ann})
for ann in annotation_contexts
).execute_annotation()

View File

@@ -91,9 +91,9 @@ def test_re_sub(Script, environment):
names = run("import re; re.sub('a', 'a', 'f')")
if environment.version_info.major == 2:
assert names == {'str', 'unicode'}
assert names == {'str'}
else:
assert names == {'str', 'bytes'}
assert names == {'str'}
# This param is missing because of overloading.
names = run("import re; re.sub('a', 'a')")