diff options
| -rw-r--r-- | tests/unittest_inference.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/tests/unittest_inference.py b/tests/unittest_inference.py index e580ee2a..c107eadb 100644 --- a/tests/unittest_inference.py +++ b/tests/unittest_inference.py @@ -5494,14 +5494,19 @@ def test_property_inference(): def test(self): return 42 + @test.setter + def test(self, value): + return "banco" + A.test #@ A().test #@ A.test.fget(A) #@ + A.test.fset(A, "a_value") #@ A.test.setter #@ A.test.getter #@ A.test.deleter #@ """ - prop, prop_result, prop_fget_result, prop_setter, prop_getter, prop_deleter = extract_node( + prop, prop_result, prop_fget_result, prop_fset_result, prop_setter, prop_getter, prop_deleter = extract_node( code ) @@ -5519,6 +5524,10 @@ def test_property_inference(): assert isinstance(inferred, nodes.Const) assert inferred.value == 42 + inferred = next(prop_fset_result.infer()) + assert isinstance(inferred, nodes.Const) + assert inferred.value == "banco" + for prop_func in prop_setter, prop_getter, prop_deleter: inferred = next(prop_func.infer()) assert isinstance(inferred, nodes.FunctionDef) |