Add _has_acb_theta function

Author: oscarbenjamin

src/flint/__init__.py

@@ -50,6 +50,23 @@
 
 __version__ = "0.8.0"
 
+
+def _flint_version_at_least(major: int, minor: int) -> bool:
+    version_parts = __FLINT_VERSION__.split(".")
+    if len(version_parts) < 2:
+        return False
+    try:
+        current_major = int(version_parts[0])
+        current_minor = int(version_parts[1])
+    except ValueError:
+        return False
+    return (current_major, current_minor) >= (major, minor)
+
+
+def _has_acb_theta() -> bool:
+    return _flint_version_at_least(3, 1)
+
+
 __all__ = [
     "ctx",
     "fmpz",

src/flint/test/test_acb_mat.py

@@ -3,7 +3,7 @@
 import sys
 from unittest.mock import patch
 
-from flint import acb, acb_mat, arb_mat, ctx, fmpq_mat, fmpz_mat
+from flint import _has_acb_theta, acb, acb_mat, arb_mat, ctx, fmpq_mat, fmpz_mat
 from flint.test.helpers import is_close_acb, is_close_acb_mat as is_close, is_close_arb_mat, raises
 
 
@@ -304,10 +304,13 @@ def test_acb_mat_eig_theta_and_helper() -> None:
     assert acb_mat(0, 0).eig() == []
 
     tau = acb_mat([[1j]])
-    theta_vals = acb_mat.theta(tau, acb_mat([[0]]))
-    assert isinstance(theta_vals, acb_mat)
-    assert theta_vals.nrows() == 1
-    assert theta_vals.ncols() == 4
+    if _has_acb_theta():
+        theta_vals = acb_mat.theta(tau, acb_mat([[0]]))
+        assert isinstance(theta_vals, acb_mat)
+        assert theta_vals.nrows() == 1
+        assert theta_vals.ncols() == 4
+    else:
+        assert raises(lambda: acb_mat.theta(tau, acb_mat([[0]])), NotImplementedError)
 
     with patch.dict(sys.modules, {"flint.types.acb_theta": None}):
         assert raises(lambda: acb_mat.theta(tau, acb_mat([[0]])), NotImplementedError)

src/flint/test/test_acb_theta.py

@@ -1,19 +1,11 @@
 from __future__ import annotations
 
-from flint import acb, acb_mat
+from flint import _has_acb_theta, acb, acb_mat
 from flint.test.helpers import is_close_acb_mat as is_close, raises
 
 
-def _acb_theta_importable() -> bool:
-    try:
-        from flint.types.acb_theta import acb_theta as _  # noqa: F401
-        return True
-    except ImportError:
-        return False
-
-
 def test_acb_theta_basic() -> None:
-    if not _acb_theta_importable():
+    if not _has_acb_theta():
         return
 
     from flint.types.acb_theta import acb_theta
@@ -35,7 +27,7 @@ def test_acb_theta_basic() -> None:
 
 
 def test_acb_theta_shape_assertions() -> None:
-    if not _acb_theta_importable():
+    if not _has_acb_theta():
         return
 
     from flint.types.acb_theta import acb_theta