diff options
-rw-r--r-- | astroid/tests/unittest_brain.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/astroid/tests/unittest_brain.py b/astroid/tests/unittest_brain.py index dba64bce..818f3a23 100644 --- a/astroid/tests/unittest_brain.py +++ b/astroid/tests/unittest_brain.py @@ -1429,8 +1429,8 @@ class TestFunctoolsPartial: ''') for node in ast_nodes: inferred = next(node.infer()) - assert isinstance(inferred, astroid.Instance) - assert inferred.qname() == 'functools.partial' + assert isinstance(inferred, (astroid.FunctionDef, astroid.Instance)) + assert inferred.qname() in ('functools.partial', 'functools.partial.newfunc') def test_inferred_partial_function_calls(self): ast_nodes = astroid.extract_node(''' |