mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-13 12:51:16 +00:00
python: Add unit test for analytics API
This is a way to validate that the GObject Annotations are correct Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/7297>
This commit is contained in:
parent
c0c0615964
commit
07205d036a
2 changed files with 135 additions and 0 deletions
|
@ -4,6 +4,7 @@ tests = [
|
|||
['Test gst', 'test_gst.py'],
|
||||
['Test fundamentals', 'test_types.py'],
|
||||
['Test plugins', 'test_plugin.py'],
|
||||
['Test analytics', 'test_analytics.py'],
|
||||
]
|
||||
|
||||
pluginsdirs = []
|
||||
|
|
134
subprojects/gst-python/testsuite/test_analytics.py
Normal file
134
subprojects/gst-python/testsuite/test_analytics.py
Normal file
|
@ -0,0 +1,134 @@
|
|||
# -*- Mode: Python -*-
|
||||
# vi:si:et:sw=4:sts=4:ts=4
|
||||
#
|
||||
# gst-python - Python bindings for GStreamer
|
||||
# Copyright (C) 2024 Collabora Ltd
|
||||
# Author: Olivier Crête <olivier.crete@collabora.com>
|
||||
#
|
||||
# This library is free software; you can redistribute it and/or
|
||||
# modify it under the terms of the GNU Lesser General Public
|
||||
# License as published by the Free Software Foundation; either
|
||||
# version 2.1 of the License, or (at your option) any later version.
|
||||
#
|
||||
# This library is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
# Lesser General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Lesser General Public
|
||||
# License along with this library; if not, write to the Free Software
|
||||
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
|
||||
import overrides_hack
|
||||
overrides_hack
|
||||
|
||||
from common import TestCase
|
||||
import unittest
|
||||
import sys
|
||||
|
||||
import gi
|
||||
gi.require_version("GLib", "2.0")
|
||||
gi.require_version("Gst", "1.0")
|
||||
gi.require_version("GstAnalytics", "1.0")
|
||||
from gi.repository import GLib
|
||||
from gi.repository import Gst
|
||||
from gi.repository import GstAnalytics
|
||||
Gst.init(None)
|
||||
|
||||
|
||||
class TestAnalyticsODMtd(TestCase):
|
||||
def test(self):
|
||||
buf = Gst.Buffer()
|
||||
self.assertIsNotNone(buf)
|
||||
|
||||
meta = GstAnalytics.buffer_add_analytics_relation_meta(buf)
|
||||
self.assertIsNotNone(meta)
|
||||
|
||||
m2 = GstAnalytics.buffer_get_analytics_relation_meta(buf)
|
||||
self.assertEqual(meta, m2)
|
||||
|
||||
qk = GLib.quark_from_string("testQuark")
|
||||
|
||||
(ret, mtd) = meta.add_od_mtd(qk, 10, 20, 30, 40, 0.3)
|
||||
self.assertTrue(ret)
|
||||
self.assertIsNotNone(mtd)
|
||||
|
||||
(ret, mtd) = meta.get_od_mtd(0)
|
||||
self.assertTrue(ret)
|
||||
self.assertIsNotNone(mtd)
|
||||
|
||||
# Ensure there is no mtd 1, only 0
|
||||
(ret, _) = meta.get_mtd(1, GstAnalytics.MTD_TYPE_ANY)
|
||||
self.assertFalse(ret)
|
||||
|
||||
# The is only one od mtd
|
||||
(ret, _) = meta.get_od_mtd(1)
|
||||
self.assertFalse(ret)
|
||||
|
||||
# There is no Class mtd
|
||||
(ret, _) = meta.get_cls_mtd(0)
|
||||
self.assertFalse(ret)
|
||||
|
||||
# meta and m2 should return the same tuple
|
||||
self.assertEqual(meta.get_od_mtd(0)[1].get_location(),
|
||||
m2.get_od_mtd(0)[1].get_location())
|
||||
|
||||
self.assertEqual(mtd.get_obj_type(), qk)
|
||||
|
||||
location = meta.get_od_mtd(0)[1].get_location()
|
||||
self.assertEqual(location[1], 10)
|
||||
self.assertEqual(location[2], 20)
|
||||
self.assertEqual(location[3], 30)
|
||||
self.assertEqual(location[4], 40)
|
||||
self.assertAlmostEqual(location[5], 0.3, 3)
|
||||
|
||||
|
||||
class TestAnalyticsClsMtd(TestCase):
|
||||
def test(self):
|
||||
buf = Gst.Buffer()
|
||||
self.assertIsNotNone(buf)
|
||||
|
||||
meta = GstAnalytics.buffer_add_analytics_relation_meta(buf)
|
||||
self.assertIsNotNone(meta)
|
||||
|
||||
qks = (GLib.quark_from_string("q1"),
|
||||
GLib.quark_from_string("q2"),
|
||||
GLib.quark_from_string("q3"))
|
||||
|
||||
(ret, mtd) = meta.add_cls_mtd([0.1, 0.2, 0.3], qks)
|
||||
self.assertTrue(ret)
|
||||
self.assertIsNotNone(mtd)
|
||||
|
||||
cnt = mtd.get_length()
|
||||
self.assertEqual(cnt, 3)
|
||||
|
||||
for i in range(cnt):
|
||||
self.assertEqual(mtd.get_index_by_quark(qks[i]), i)
|
||||
self.assertAlmostEqual(mtd.get_level(i), (i + 1) / 10, 7)
|
||||
self.assertEqual(mtd.get_quark(i), qks[i])
|
||||
|
||||
|
||||
class TestAnalyticsTrackingMtd(TestCase):
|
||||
def test(self):
|
||||
buf = Gst.Buffer()
|
||||
self.assertIsNotNone(buf)
|
||||
|
||||
meta = GstAnalytics.buffer_add_analytics_relation_meta(buf)
|
||||
self.assertIsNotNone(meta)
|
||||
|
||||
(ret, mtd) = meta.add_tracking_mtd(1, 10)
|
||||
self.assertTrue(ret)
|
||||
rets = mtd.get_info()
|
||||
self.assertFalse(rets.tracking_lost)
|
||||
self.assertEqual(rets.tracking_first_seen, 10)
|
||||
self.assertEqual(rets.tracking_last_seen, 10)
|
||||
|
||||
mtd.update_last_seen(20)
|
||||
|
||||
rets = mtd.get_info()
|
||||
self.assertEqual(rets.tracking_first_seen, 10)
|
||||
self.assertEqual(rets.tracking_last_seen, 20)
|
||||
|
||||
mtd.set_lost()
|
||||
rets = mtd.get_info()
|
||||
self.assertTrue(rets.tracking_lost)
|
Loading…
Reference in a new issue