From 8d0083c4aafd72dcddd9190d5323406d10171e17 Mon Sep 17 00:00:00 2001 From: wyzula-jan <133381102+wyzula-jan@users.noreply.github.com> Date: Tue, 23 Jan 2024 14:42:11 +0100 Subject: [PATCH] test: fix test_bec_monitor_scatter2D.py database init test change to check defaultdict --- tests/test_bec_monitor_scatter2D.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/tests/test_bec_monitor_scatter2D.py b/tests/test_bec_monitor_scatter2D.py index 2855f778..ca3b7ad8 100644 --- a/tests/test_bec_monitor_scatter2D.py +++ b/tests/test_bec_monitor_scatter2D.py @@ -1,4 +1,5 @@ # pylint: disable=missing-module-docstring, missing-function-docstring +from collections import defaultdict import pytest from unittest.mock import MagicMock @@ -89,10 +90,22 @@ def test_initialization(monitor_2Dscatter, config, number_of_plots): ) def test_database_initialization(monitor_2Dscatter, config): monitor_2Dscatter.on_config_update(config) + # Check if the database is a defaultdict + assert isinstance(monitor_2Dscatter.database, defaultdict) + for axis_dict in monitor_2Dscatter.database.values(): + assert isinstance(axis_dict, defaultdict) + for signal_list in axis_dict.values(): + assert isinstance(signal_list, defaultdict) + + # Access the elements for plot_config in config["waveform2D"]: plot_name = plot_config["plot_name"] - assert plot_name in monitor_2Dscatter.database - assert all(axis in monitor_2Dscatter.database[plot_name] for axis in ["x", "y", "z"]) + + for axis in ["x", "y", "z"]: + for signal in plot_config["signals"][axis]: + signal_name = signal["name"] + assert not monitor_2Dscatter.database[plot_name][axis][signal_name] + assert isinstance(monitor_2Dscatter.database[plot_name][axis][signal_name], list) @pytest.mark.parametrize(