assert_log_level fails if higher-level log than expected, string matching contains_str, AssertLogLevel context

Author: momchil-flex

CHANGELOG.md

@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - `ModeData.dispersion` and `ModeSolverData.dispersion` are calculated together with the group index.
 - String matching feature `contains_str` to `assert_log_level` testing utility.
 - Warning in automatic grid generation if a structure has a non-zero size along a given direction that is too small compared to a single mesh step.
+- `assert_log_level` adds string matching with `contains_str` and ensures no higher log level recorded than expected.
+- `AssertLogLevel` context manager for testing log level and automatically clearing capture.
 
 ### Changed
 - `jax` and `jaxlib` versions bumped to `0.4.*`.

tests/test_components/test_medium.py

@@ -5,7 +5,7 @@
 import matplotlib.pyplot as plt
 import tidy3d as td
 from tidy3d.exceptions import ValidationError, SetupError
-from ..utils import assert_log_level, log_capture
+from ..utils import assert_log_level, log_capture, AssertLogLevel
 from typing import Dict
 
 MEDIUM = td.Medium()
@@ -338,11 +338,12 @@ def test_gain_medium(log_capture):
         _ = td.Drude(eps_inf=0.04, coeffs=[(1, -2)])
 
     # anisotropic medium, warn allow_gain is ignored
-    _ = td.AnisotropicMedium(xx=td.Medium(), yy=mL, zz=mS, allow_gain=True)
-    assert_log_level(log_capture, "WARNING")
 
-    _ = td.AnisotropicMedium(xx=td.Medium(), yy=mL, zz=mS, allow_gain=False)
-    assert_log_level(log_capture, "WARNING")
+    with AssertLogLevel(log_capture, "WARNING"):
+        _ = td.AnisotropicMedium(xx=td.Medium(), yy=mL, zz=mS, allow_gain=True)
+
+    with AssertLogLevel(log_capture, "WARNING"):
+        _ = td.AnisotropicMedium(xx=td.Medium(), yy=mL, zz=mS, allow_gain=False)
 
 
 def test_medium2d(log_capture):