summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorTomi Valkeinen <tomi.valkeinen@ideasonboard.com>2022-05-18 16:13:18 +0300
committerLaurent Pinchart <laurent.pinchart@ideasonboard.com>2022-05-18 17:52:28 +0300
commit7ac7e0e119edfd9bdba5bafec48512338598eea4 (patch)
tree0c49a6172182f5612b2a40796cf25402d73be07a
parent518f46ff0fc59743d490ae646c34fd35f4801984 (diff)
py: unittests: Make typechecker happy
Add some annotations and self.assertIsNotNone() calls to remove the typechecker warnings. Signed-off-by: Tomi Valkeinen <tomi.valkeinen@ideasonboard.com> Reviewed-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com> Reviewed-by: Kieran Bingham <kieran.bingham@ideasonboard.com> Signed-off-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
-rwxr-xr-xtest/py/unittests.py13
1 files changed, 11 insertions, 2 deletions
diff --git a/test/py/unittests.py b/test/py/unittests.py
index 4c214f0a..7dede33b 100755
--- a/test/py/unittests.py
+++ b/test/py/unittests.py
@@ -10,6 +10,7 @@ import libcamera as libcam
import os
import selectors
import time
+import typing
import unittest
import weakref
@@ -70,6 +71,9 @@ class SimpleTestMethods(BaseTestCase):
class CameraTesterBase(BaseTestCase):
+ cm: typing.Any
+ cam: typing.Any
+
def setUp(self):
self.cm = libcam.CameraManager.singleton()
self.cam = next((cam for cam in self.cm.cameras if 'platform/vimc' in cam.id), None)
@@ -131,6 +135,7 @@ class AllocatorTestMethods(CameraTesterBase):
wr_allocator = weakref.ref(allocator)
buffers = allocator.buffers(stream)
+ self.assertIsNotNone(buffers)
buffers = None
buffer = allocator.buffers(stream)[0]
@@ -166,6 +171,8 @@ class SimpleCaptureMethods(CameraTesterBase):
streamconfig = camconfig.at(0)
fmts = streamconfig.formats
+ self.assertIsNotNone(fmts)
+ fmts = None
ret = cam.configure(camconfig)
self.assertZero(ret)
@@ -225,6 +232,8 @@ class SimpleCaptureMethods(CameraTesterBase):
streamconfig = camconfig.at(0)
fmts = streamconfig.formats
+ self.assertIsNotNone(fmts)
+ fmts = None
ret = cam.configure(camconfig)
self.assertZero(ret)
@@ -348,9 +357,9 @@ if __name__ == '__main__':
gc.unfreeze()
gc.collect()
- obs_after = get_all_objects([obs_before])
+ obs_after = get_all_objects([obs_before]) # type: ignore
- before = create_type_count_map(obs_before)
+ before = create_type_count_map(obs_before) # type: ignore
after = create_type_count_map(obs_after)
leaks = diff_type_count_maps(before, after)