jedi: Update tests for stalled, done.

pybricks · Dec 2, 2022 · 15dbfb0 · 15dbfb0
1 parent 8fb9afb
commit 15dbfb0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/jedi/tests/test_get_signature.py b/jedi/tests/test_get_signature.py
@@ -602,9 +602,9 @@ def _get_method_signature(module: str, type: str, method: str) -> SignatureHelp:
         ],
     ),
     pytest.param("pybricks.pupdevices", "Motor", "dc", [(["duty: Number"], "None")]),
-    pytest.param("pybricks.pupdevices", "Motor", "control.done", [([], "bool")]),
-    pytest.param("pybricks.pupdevices", "Motor", "control.stalled", [([], "bool")]),
-    pytest.param("pybricks.pupdevices", "Motor", "control.load", [([], "int")]),
+    pytest.param("pybricks.pupdevices", "Motor", "done", [([], "bool")]),
+    pytest.param("pybricks.pupdevices", "Motor", "stalled", [([], "bool")]),
+    pytest.param("pybricks.pupdevices", "Motor", "load", [([], "int")]),
     pytest.param(
         "pybricks.pupdevices",
         "Motor",
@@ -853,6 +853,8 @@ def _get_method_signature(module: str, type: str, method: str) -> SignatureHelp:
         "pybricks.robotics", "DriveBase", "state", [([], "Tuple[int, int, int, int]")]
     ),
     pytest.param("pybricks.robotics", "DriveBase", "reset", [([], "None")]),
+    pytest.param("pybricks.robotics", "DriveBase", "done", [([], "bool")]),
+    pytest.param("pybricks.robotics", "DriveBase", "stalled", [([], "bool")]),
 ]