Bug fixes

tests/test_contexts.py

@@ -78,7 +78,7 @@ def test_nt_sim_context_bad_inits(self):
 
 
 def test_sim_context():
-    straxen.contexts.xenon1t_simulation()
+    wfsim.contexts.xenon1t_simulation()
 
 
 @skipIf(not straxen.utilix_is_configured(), "No db access, cannot test!")

tests/test_wfsim.py

@@ -51,7 +51,7 @@ def test_sim_1T():
                     **conf_1t,
                 ),
                 **straxen.legacy.x1t_common_config),
-            **straxen.legacy.contexts_1t.get_x1t_context_config(),
+            **straxen.legacy.get_x1t_context_config(),
         )
         st.register(wfsim.RawRecordsFromFax1T)
         log.debug(f'Setting testing config {testing_config_1t}')

wfsim/__init__.py

@@ -11,3 +11,5 @@
 
 from .load_resource import *
 from .utils import *
+
+from .contexts import *

wfsim/contexts.py

@@ -2,6 +2,7 @@
 from immutabledict import immutabledict
 import strax
 import straxen
+import wfsim
 from .strax_interface import RawRecordsFromFax1T
 
 
@@ -280,8 +281,11 @@ def xenonnt_simulation(
 def xenon1t_simulation(output_folder="./strax_data"):
     st = strax.Context(
         storage=strax.DataDirectory(output_folder),
-        config=dict(fax_config="fax_config_1t.json", detector="XENON1T", **straxen.contexts.x1t_common_config),
-        **straxen.contexts.get_x1t_context_config(),
+        config=dict(
+            fax_config="fax_config_1t.json", detector="XENON1T",
+            **straxen.legacy.x1t_common_config,
+        ),
+        **straxen.legacy.get_x1t_context_config(),
     )
     st.register(RawRecordsFromFax1T)
     st.deregister_plugins_with_missing_dependencies()