Change some tests slightly
This commit is contained in:
1 parent
696df90daf
commit
1820aa9476
2 files changed
+2
-2
No files matched your search
@@ -134,7 +134,7 @@ def test_infer_on_non_name(Script):
|
||||
def test_infer_on_generator(Script, environment):
|
||||
script = Script('def x(): yield 1\ny=x()\ny')
|
||||
def_, = script.infer()
|
||||
assert def_.name == 'Generator'
|
||||
assert def_.name == 'GeneratorType'
|
||||
def_, = script.infer(only_stubs=True)
|
||||
assert def_.name == 'Generator'
|
||||
|
||||
|
||||
Reference in new issue
Block a user