Test StatsManager metric registration/setting/getting with a set of pre-defined key-value pairs (metric_dict).
()
| 44 | |
| 45 | |
| 46 | def test_metrics(): |
| 47 | """Test StatsManager metric registration/setting/getting with a set of pre-defined |
| 48 | key-value pairs (metric_dict). |
| 49 | """ |
| 50 | metric_dict = {"some_metric": 1.2345, "another_metric": 6.7890} |
| 51 | metric_keys = list(metric_dict.keys()) |
| 52 | |
| 53 | stats = StatsManager() |
| 54 | frame_key = 100 |
| 55 | assert not stats.is_save_required() |
| 56 | |
| 57 | stats.register_metrics(metric_keys) |
| 58 | |
| 59 | assert not stats.is_save_required() |
| 60 | |
| 61 | assert not stats.metrics_exist(frame_key, metric_keys) |
| 62 | assert stats.get_metrics(frame_key, metric_keys) == [None] * len(metric_keys) |
| 63 | |
| 64 | stats.set_metrics(frame_key, metric_dict) |
| 65 | |
| 66 | assert stats.is_save_required() |
| 67 | |
| 68 | assert stats.metrics_exist(frame_key, metric_keys) |
| 69 | assert stats.metrics_exist(frame_key, metric_keys[1:]) |
| 70 | |
| 71 | assert stats.get_metrics(frame_key, metric_keys) == [ |
| 72 | metric_dict[metric_key] for metric_key in metric_keys |
| 73 | ] |
| 74 | |
| 75 | |
| 76 | def test_detector_metrics(test_video_file): |
nothing calls this directly
no test coverage detected