Merge pull request #75 from Ericgig/misc.familly

Add entry point for qutip-family

setup.cfg

@@ -38,6 +38,10 @@ setup_requires =
 [options.packages.find]
 where = src
 
+[options.entry_points]
+qutip.family =
+    qutip_jax = qutip_jax.family
+
 [options.extras_require]
 tests =
     pytest>=6.0

src/qutip_jax/family.py

@@ -0,0 +1,8 @@
+"""QuTiP family package entry point."""
+
+from . import __version__
+
+
+def version():
+    """Return information to include in qutip.about()."""
+    return "qutip-jax", __version__

tests/test_family.py

@@ -0,0 +1,12 @@
+""" Tests for qutip_jax.family. """
+
+import re
+
+from qutip_jax import family
+
+
+class TestVersion:
+    def test_version(self):
+        pkg, version = family.version()
+        assert pkg == "qutip-jax"
+        assert re.match(r"\d+\.\d+\.\d+.*", version)