extend test coverage

materialsproject · Jun 12, 2024 · b1e8e9e · b1e8e9e
1 parent 58d247c
commit b1e8e9e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/core/test_units.py b/tests/core/test_units.py
@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+import warnings
+
 import pytest
 from numpy.testing import assert_array_equal
 from pytest import approx
@@ -109,10 +111,15 @@ def test_memory(self):
 
     def test_deprecated_memory(self):
         # TODO: remove after 2025-01-01
-        for unit in ("mb", "Mb"):
+        for unit in ("Kb", "kb", "Mb", "mb", "Gb", "gb", "Tb", "tb"):
             with pytest.warns(DeprecationWarning, match=f"Unit {unit} is deprecated"):
                 Memory(1, unit)
 
+        with warnings.catch_warnings():
+            warnings.simplefilter("error")
+            for unit in ("KB", "MB", "GB", "TB"):
+                Memory(1, unit)
+
     def test_unitized(self):
         @unitized("eV")
         def func1():