Add type hints

nvaccess · Jan 7, 2025 · b01ad17 · b01ad17
1 parent c24ad0f
commit b01ad17
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit/test_bdDetect.py b/tests/unit/test_bdDetect.py
@@ -45,7 +45,7 @@ def test_addUsbDevice(self):
 
 		registrar = bdDetect.DriverRegistrar(albatross.BrailleDisplayDriver.name)
 
-		def matchFunc(match):
+		def matchFunc(match: bdDetect.DeviceMatch) -> bool:
 			return match.deviceInfo.get("busReportedDeviceDescription") == albatross.driver.BUS_DEVICE_DESC
 
 		registrar.addUsbDevice(
@@ -66,7 +66,7 @@ def test_addUsbDevices(self):
 
 		registrar = bdDetect.DriverRegistrar(albatross.BrailleDisplayDriver.name)
 
-		def matchFunc(match):
+		def matchFunc(match: bdDetect.DeviceMatch) -> bool:
 			return match.deviceInfo.get("busReportedDeviceDescription") == albatross.driver.BUS_DEVICE_DESC
 
 		fakeVidAndPid = "VID_0403&PID_6002"
@@ -94,7 +94,7 @@ def test_addBluetoothDevices(self):
 
 		registrar = bdDetect.DriverRegistrar(albatross.BrailleDisplayDriver.name)
 
-		def matchFunc(match):
+		def matchFunc(match: bdDetect.DeviceMatch) -> bool:
 			return True
 
 		registrar.addBluetoothDevices(matchFunc)