Capturing warnings.

Author: vitenti

firecrown/app/sacc/_load.py

@@ -1,6 +1,9 @@
 """Load command for SACC files."""
 
 import dataclasses
+import contextlib
+import io
+import warnings
 from pathlib import Path
 from typing import Annotated
 
@@ -41,7 +44,14 @@ def _load_sacc_file(self) -> None:
         try:
             if not self.sacc_file.exists():
                 raise typer.BadParameter(f"SACC file not found: {self.sacc_file}")
-            self.sacc_data = factories.load_sacc_data(self.sacc_file.as_posix())
+
+            with (
+                contextlib.redirect_stdout(io.StringIO()),
+                contextlib.redirect_stderr(io.StringIO()),
+                warnings.catch_warnings(),
+            ):
+                warnings.simplefilter("ignore")
+                self.sacc_data = factories.load_sacc_data(self.sacc_file.as_posix())
         except Exception as e:
             self.console.print(f"[bold red]Failed to load SACC file:[/bold red] {e}")
             raise

tests/likelihood/gauss_family/test_const_gaussianPM.py

@@ -214,9 +214,11 @@ def test_compute_chisq_fails_before_read(trivial_stats):
     """Note that the error message from the direct call to compute_chisq notes
     that update() must be called; this can only be called after read()."""
     likelihood = ConstGaussianPM(statistics=trivial_stats)
-    with pytest.raises(
-        AssertionError,
-        match=re.escape("update() must be called before compute_chisq()"),
+    with (
+        pytest.raises(
+            AssertionError,
+            match=re.escape("update() must be called before compute_chisq()"),
+        ),
     ):
         _ = likelihood.compute_chisq(ModelingTools())
 
@@ -225,7 +227,11 @@ def test_compute_chisq(trivial_stats, sacc_data_for_trivial_stat, trivial_params
     likelihood = ConstGaussianPM(statistics=trivial_stats)
     likelihood.read(sacc_data_for_trivial_stat)
     likelihood.update(trivial_params)
-    assert likelihood.compute_chisq(ModelingTools()) == 2.0
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        assert likelihood.compute_chisq(ModelingTools()) == 2.0
 
 
 def test_deprecated_compute(trivial_stats, sacc_data_for_trivial_stat, trivial_params):
@@ -256,7 +262,11 @@ def compute(
     likelihood.compute_theory_vector = compute_theory_vector  # type: ignore
     likelihood.compute = compute  # type: ignore
 
-    assert likelihood.compute_chisq(ModelingTools()) == 2.0
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        assert likelihood.compute_chisq(ModelingTools()) == 2.0
 
 
 def test_required_parameters(trivial_stats, sacc_data_for_trivial_stat, trivial_params):
@@ -282,7 +292,11 @@ def test_reset(trivial_stats, sacc_data_for_trivial_stat, trivial_params):
     likelihood.read(sacc_data_for_trivial_stat)
     likelihood.update(trivial_params)
     assert not trivial_stats[0].computed_theory_vector
-    assert likelihood.compute_loglike(ModelingTools()) == -1.0
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        assert likelihood.compute_loglike(ModelingTools()) == -1.0
     assert trivial_stats[0].computed_theory_vector
     likelihood.reset()
     assert not trivial_stats[0].computed_theory_vector
@@ -307,7 +321,11 @@ def test_using_good_sacc(
     likelihood.read(sacc_data_for_trivial_stat)
     params = firecrown.parameters.ParamsMap(mean=10.5)
     likelihood.update(params)
-    chisq = likelihood.compute_chisq(tools_with_vanilla_cosmology)
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        chisq = likelihood.compute_chisq(tools_with_vanilla_cosmology)
     assert isinstance(chisq, float)
     assert chisq > 0.0
 
@@ -331,15 +349,23 @@ def test_make_realization_chisq(
     likelihood.read(sacc_data_for_trivial_stat)
     params = firecrown.parameters.ParamsMap(mean=10.5)
     likelihood.update(params)
-    likelihood.compute_chisq(tools_with_vanilla_cosmology)
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        likelihood.compute_chisq(tools_with_vanilla_cosmology)
 
     new_sacc = likelihood.make_realization(sacc_data_for_trivial_stat)
 
     new_likelihood = ConstGaussianPM(statistics=[TrivialStatistic()])
     new_likelihood.read(new_sacc)
     params = firecrown.parameters.ParamsMap(mean=10.5)
     new_likelihood.update(params)
-    chisq = new_likelihood.compute_chisq(tools_with_vanilla_cosmology)
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        chisq = new_likelihood.compute_chisq(tools_with_vanilla_cosmology)
 
     # The new likelihood chisq is distributed as a chi-squared with 3 degrees of
     # freedom. We want to check that the new chisq is within the 1-10^-6 quantile
@@ -358,7 +384,11 @@ def test_make_realization_chisq_mean(
     likelihood.read(sacc_data_for_trivial_stat)
     params = firecrown.parameters.ParamsMap(mean=10.5)
     likelihood.update(params)
-    likelihood.compute_chisq(tools_with_vanilla_cosmology)
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        likelihood.compute_chisq(tools_with_vanilla_cosmology)
 
     chisq_list = []
     for _ in range(1000):
@@ -368,7 +398,11 @@ def test_make_realization_chisq_mean(
         new_likelihood.read(new_sacc)
         params = firecrown.parameters.ParamsMap(mean=10.5)
         new_likelihood.update(params)
-        chisq = new_likelihood.compute_chisq(tools_with_vanilla_cosmology)
+        with pytest.warns(
+            UserWarning,
+            match=re.escape("inverse covariance correction has not yet been computed."),
+        ):
+            chisq = new_likelihood.compute_chisq(tools_with_vanilla_cosmology)
         chisq_list.append(chisq)
 
     # The new likelihood chisq is distributed as a chi-squared with 3 degrees of
@@ -387,7 +421,11 @@ def test_make_realization_data_vector(
     likelihood.read(sacc_data_for_trivial_stat)
     params = firecrown.parameters.ParamsMap(mean=10.5)
     likelihood.update(params)
-    likelihood.compute_chisq(tools_with_vanilla_cosmology)
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        likelihood.compute_chisq(tools_with_vanilla_cosmology)
 
     data_vector_list = []
     for _ in range(1000):
@@ -430,7 +468,11 @@ def test_make_realization_no_noise(
     likelihood.read(sacc_data_for_trivial_stat)
     params = firecrown.parameters.ParamsMap(mean=10.5)
     likelihood.update(params)
-    likelihood.compute_chisq(tools_with_vanilla_cosmology)
+    with pytest.warns(
+        UserWarning,
+        match=re.escape("inverse covariance correction has not yet been computed."),
+    ):
+        likelihood.compute_chisq(tools_with_vanilla_cosmology)
 
     new_sacc = likelihood.make_realization(sacc_data_for_trivial_stat, add_noise=False)
 
@@ -561,8 +603,8 @@ def __init__(self, statistic):
         self.statistic = statistic
 
 
-@pytest.fixture
-def minimal_const_gaussian_PM():
+@pytest.fixture(name="minimal_const_gaussian_PM")
+def fixture_minimal_const_gaussian_PM() -> ConstGaussianPM:
     # Create minimal valid statistics for the class.
     z = np.array([0.1, 0.2, 0.3])
     dndz = np.array([1.0, 2.0, 3.0])
@@ -601,6 +643,9 @@ def minimal_const_gaussian_PM():
     return likelihood
 
 
+# pylint: disable=protected-access
+
+
 def test_precomputed_warning(minimal_const_gaussian_PM):
     # Check that running the precomputation twice gives a warning.
     minimal_const_gaussian_PM._generate_maps()
@@ -818,3 +863,6 @@ def __init__(self):
 
     with pytest.raises(StopIteration, match="missing attributes"):
         likelihood._collect_data_vectors()
+
+
+# pylint enable=protected-access