diff --git a/test_text_spotting/test_deepsolo_runner.py b/test_text_spotting/test_deepsolo_runner.py index 181ac714..78f7763e 100644 --- a/test_text_spotting/test_deepsolo_runner.py +++ b/test_text_spotting/test_deepsolo_runner.py @@ -7,6 +7,7 @@ import geopandas as gpd import pandas as pd import pytest +from deepsolo.config import get_cfg from detectron2.engine import DefaultPredictor from detectron2.structures.instances import Instances @@ -70,6 +71,11 @@ def runner_run_all(init_runner, mock_response): return runner +def test_get_cfg(): + cfg = get_cfg() + assert "TEMPERATURE" in cfg["MODEL"]["TRANSFORMER"].keys() + + def test_deepsolo_init(init_dataframes): parent_df, patch_df = init_dataframes runner = DeepSoloRunner( diff --git a/test_text_spotting/test_dptext_runner.py b/test_text_spotting/test_dptext_runner.py index 056519a2..06e224fe 100644 --- a/test_text_spotting/test_dptext_runner.py +++ b/test_text_spotting/test_dptext_runner.py @@ -9,6 +9,7 @@ import pytest from detectron2.engine import DefaultPredictor from detectron2.structures.instances import Instances +from dptext_detr.config import get_cfg from mapreader import DPTextDETRRunner from mapreader.load import MapImages @@ -70,6 +71,12 @@ def runner_run_all(init_runner, mock_response): return runner +def test_get_cfg(): + cfg = get_cfg() + assert "USE_POLYGON" in cfg["MODEL"]["TRANSFORMER"].keys() + assert "DET_ONLY" in cfg["TEST"].keys() + + def test_dptext_init(init_dataframes): parent_df, patch_df = init_dataframes runner = DPTextDETRRunner( diff --git a/test_text_spotting/test_maptext_runner.py b/test_text_spotting/test_maptext_runner.py index a06a611a..3c2bbab3 100644 --- a/test_text_spotting/test_maptext_runner.py +++ b/test_text_spotting/test_maptext_runner.py @@ -73,8 +73,8 @@ def runner_run_all(init_runner, mock_response): def test_get_cfg(): cfg = get_cfg() - print(cfg, flush=True) - assert cfg["MODEL"]["TRANSFORMER"].get("TEMPERATURE") + assert "TEMPERATURE" in cfg["MODEL"]["TRANSFORMER"].keys() + assert "FROZEN" in cfg["MODEL"]["TRANSFORMER"].keys() def test_maptext_init(init_dataframes):