From 4ea7981680d4c7d0e9f5792af0ef6a0980b8fb1b Mon Sep 17 00:00:00 2001 From: Eric Masseran Date: Fri, 29 Aug 2025 18:37:51 +0200 Subject: [PATCH] Add complete test --- test/test_inference/test_signature.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/test/test_inference/test_signature.py b/test/test_inference/test_signature.py index fca28ea2..acd6eafb 100644 --- a/test/test_inference/test_signature.py +++ b/test/test_inference/test_signature.py @@ -771,6 +771,23 @@ def test_extensions_dataclass_transform_signature( assert price.name == price_type_infer +def test_dataclass_transform_complete(Script): + script = Script('''\ + @dataclass_transform + class Y(): + y: int + z = 5 + + class X(Y): + name: str + foo = 3 + + def f(x: X): + x.na''') + completion, = script.complete() + assert completion.description == 'name: str' + + @pytest.mark.parametrize( "start, start_params, include_params", dataclass_transform_cases, ids=ids )