added AM_VERSION constant, added to module

Author: flexatone

arraymap.c

@@ -2105,20 +2105,21 @@ PyInit_arraymap(void)
         return NULL;
     }
 
-    PyObject *arraymap = PyModule_Create(&arraymap_module);
+    PyObject *m = PyModule_Create(&arraymap_module);
     if (
-        !arraymap
+        !m
+        || PyModule_AddStringConstant(m, "__version__", Py_STRINGIFY(AM_VERSION))
         || PyType_Ready(&AMType)
         || PyType_Ready(&FAMIType)
         || PyType_Ready(&FAMVType)
         || PyType_Ready(&FAMType)
-        || PyModule_AddObject(arraymap, "AutoMap", (PyObject *)&AMType)
-        || PyModule_AddObject(arraymap, "FrozenAutoMap", (PyObject *)&FAMType)
-        || PyModule_AddObject(arraymap, "NonUniqueError", NonUniqueError)
+        || PyModule_AddObject(m, "AutoMap", (PyObject *)&AMType)
+        || PyModule_AddObject(m, "FrozenAutoMap", (PyObject *)&FAMType)
+        || PyModule_AddObject(m, "NonUniqueError", NonUniqueError)
     ) {
-        Py_XDECREF(arraymap);
+        Py_XDECREF(m);
         return NULL;
     }
-    return arraymap;
+    return m;
 }
 

setup.py

@@ -3,6 +3,10 @@
 import site
 import os
 
+
+AM_VERSION = "0.1.1"
+
+
 with open("README.md") as file:
     LONG_DESCRIPTION = file.read()
 
@@ -21,12 +25,14 @@ def get_ext_dir(*components: tp.Iterable[str]) -> tp.Sequence[str]:
     ["arraymap.c"],
     include_dirs=get_ext_dir("numpy", "core", "include"),
     library_dirs=get_ext_dir("numpy", "core", "lib"),
+    define_macros=[("AM_VERSION", AM_VERSION)],
     libraries=["npymath"],  # not including mlib at this time
 )
 
 
 setuptools.setup(
     author="Christopher Ariza, Brandt Bucher",
+    version=AM_VERSION,
     description="Dictionary-like lookup from NumPy array values to their integer positions",
     ext_modules=[extension],
     license="MIT",
@@ -35,5 +41,4 @@ def get_ext_dir(*components: tp.Iterable[str]) -> tp.Sequence[str]:
     name="arraymap",
     python_requires=">=3.7.0",
     url="https://github.com/static-frame/arraymap",
-    version="0.1.0",
 )

tasks.py

@@ -28,4 +28,4 @@ def build(context):
 
 @invoke.task(build)
 def test(context):
-    run(context, f"{sys.executable} -m pytest -v")
+    run(context, f"{sys.executable} -m pytest")

test/test_unit.py

@@ -8,6 +8,13 @@
 from arraymap import FrozenAutoMap
 from arraymap import NonUniqueError
 
+def test_version():
+    import arraymap
+    assert isinstance(arraymap.__version__, str)
+
+
+# ------------------------------------------------------------------------------
+
 
 def test_am_extend():
     am1 = AutoMap(("a", "b"))