FaSTTUBe/GST-Tensordecoder-ov_ep
37
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

python tests: Adapt analytics unit test

Browse Source 
- Adapt test to batch size moved into dims field

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/8191>
This commit is contained in:
Daniel Morin 2024-12-30 18:05:36 -05:00 committed by GStreamer Marge Bot
parent 0a343fa738
commit e272e229e8
1 changed files with 19 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
subprojects/gst-python/testsuite/test_analytics.py
View File

@ -116,6 +116,7 @@ class TestAnalyticsODMtd(TestCase):
self.assertEqual(location[4], 212) self.assertEqual(location[4], 212)
self.assertAlmostEqual(location[5], 0.3, 3) self.assertAlmostEqual(location[5], 0.3, 3)
class TestAnalyticsClsMtd(TestCase): class TestAnalyticsClsMtd(TestCase):
def test(self): def test(self):
buf = Gst.Buffer() buf = Gst.Buffer()
@ -209,30 +210,38 @@ class TestAnalyticsTensorMeta(TestCase):
self.assertIsNotNone(data) self.assertIsNotNone(data)
tensor = GstAnalytics.Tensor.new_simple(0, GstAnalytics.TensorDataType.UINT8, tensor = GstAnalytics.Tensor.new_simple(0, GstAnalytics.TensorDataType.UINT8,
1, data, data,
GstAnalytics.TensorDimOrder.ROW_MAJOR, GstAnalytics.TensorDimOrder.ROW_MAJOR,
[2, 3, 4]) [1, 2, 3, 4])
self.assertIsNotNone(tensor) self.assertIsNotNone(tensor)
self.assertEqual(tensor.id, 0) self.assertEqual(tensor.id, 0)
self.assertEqual(tensor.num_dims, 3) self.assertEqual(tensor.num_dims, 4)
self.assertEqual(tensor.batch_size, 1)
dims = tensor.get_dims() dims = tensor.get_dims()
self.assertEqual(len(dims), 3) self.assertEqual(len(dims), 4)
self.assertEqual(dims[0].size, 2) self.assertEqual(dims[0].size, 1)
self.assertEqual(dims[1].size, 3) self.assertEqual(dims[1].size, 2)
self.assertEqual(dims[2].size, 4) self.assertEqual(dims[2].size, 3)
self.assertEqual(dims[3].size, 4)
self.assertEqual(tensor.data, data) self.assertEqual(tensor.data, data)
self.assertEqual(tensor.data_type, GstAnalytics.TensorDataType.UINT8) self.assertEqual(tensor.data_type, GstAnalytics.TensorDataType.UINT8)
self.assertEqual(tensor.dims_order, GstAnalytics.TensorDimOrder.ROW_MAJOR) self.assertEqual(tensor.dims_order, GstAnalytics.TensorDimOrder.ROW_MAJOR)
data2 = Gst.Buffer.new_allocate(None, 2 * 3 * 4 * 5) data2 = Gst.Buffer.new_allocate(None, 2 * 3 * 4 * 5)
tensor2 = GstAnalytics.Tensor.new_simple(0, GstAnalytics.TensorDataType.UINT16, tensor2 = GstAnalytics.Tensor.new_simple(0, GstAnalytics.TensorDataType.UINT16,
1, data2, data2,
GstAnalytics.TensorDimOrder.ROW_MAJOR, GstAnalytics.TensorDimOrder.ROW_MAJOR,
[3, 4, 5]) [1, 3, 4, 5])
tmeta.set([tensor, tensor2]) tmeta.set([tensor, tensor2])
tmeta2 = GstAnalytics.buffer_get_tensor_meta(buf) tmeta2 = GstAnalytics.buffer_get_tensor_meta(buf)
self.assertEqual(tmeta2.num_tensors, 2) self.assertEqual(tmeta2.num_tensors, 2)
self.assertEqual(tmeta2.get(0).data, data) self.assertEqual(tmeta2.get(0).data, data)
self.assertEqual(tmeta2.get(1).data, data2) self.assertEqual(tmeta2.get(1).data, data2)
data3 = Gst.Buffer.new_allocate(None, 30)
tensor3 = GstAnalytics.Tensor.new_simple(0,
GstAnalytics.TensorDataType.UINT16,
data3,
GstAnalytics.TensorDimOrder.ROW_MAJOR,
[0, 2, 5])
self.assertIsNotNone(tensor3)