update tests for get_cfg

maps-as-data · Oct 22, 2024 · a3f7186 · a3f7186
1 parent 8653909
commit a3f7186
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git 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
@@ -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
@@ -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):