2424import numpy as np
2525
2626from climada import CONFIG
27- from climada .entity .measures .base import Measure
28- from climada .entity .measures .measure_set import MeasureSet
27+ from climada .entity ._legacy_measures .base import Measure
28+ from climada .entity ._legacy_measures .measure_set import MeasureSet
2929from climada .util .constants import ENT_DEMO_TODAY , ENT_TEMPLATE_XLS
3030
3131DATA_DIR = CONFIG .measures .test_data .dir ()
@@ -58,7 +58,7 @@ def test_add_wrong_error(self):
5858 """Test error is raised when wrong ImpactFunc provided."""
5959 meas = MeasureSet ()
6060 with self .assertLogs (
61- "climada.entity.measures .measure_set" , level = "WARNING"
61+ "climada.entity._legacy_measures .measure_set" , level = "WARNING"
6262 ) as cm :
6363 meas .append (Measure ())
6464 self .assertIn ("Input Measure's hazard type not set." , cm .output [0 ])
@@ -76,7 +76,9 @@ def test_remove_measure_pass(self):
7676 def test_remove_wrong_error (self ):
7777 """Test error is raised when invalid inputs."""
7878 meas = MeasureSet (measure_list = [Measure (name = "Mangrove" , haz_type = "FL" )])
79- with self .assertLogs ("climada.entity.measures.measure_set" , level = "INFO" ) as cm :
79+ with self .assertLogs (
80+ "climada.entity._legacy_measures.measure_set" , level = "INFO"
81+ ) as cm :
8082 meas .remove_measure (name = "Seawall" )
8183 self .assertIn ("No Measure with name Seawall." , cm .output [0 ])
8284
0 commit comments