0
0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-13 19:21:50 +02:00

test: fix test_bec_monitor_scatter2D.py database init test change to check defaultdict

This commit is contained in:
wyzula-jan
2024-01-23 14:42:11 +01:00
parent 3c143274c5
commit 8d0083c4aa

View File

@ -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(