add __version__

oconnor663 · Feb 16, 2021 · 38aca74 · 38aca74
1 parent 6fe91b9
commit 38aca74
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/lib.rs b/src/lib.rs
@@ -318,5 +318,6 @@ fn blake3(_: Python, m: &PyModule) -> PyResult<()> {
     m.add_wrapped(wrap_pyfunction!(blake3))?;
     m.add("OUT_LEN", blake3::OUT_LEN)?;
     m.add("KEY_LEN", blake3::KEY_LEN)?;
+    m.add("__version__", env!("CARGO_PKG_VERSION"))?;
     Ok(())
 }
diff --git a/tests/test_blake3.py b/tests/test_blake3.py
@@ -7,7 +7,7 @@
 import sys
 
 try:
-    from blake3 import blake3
+    from blake3 import blake3, __version__
 except ModuleNotFoundError:
     print("Run tests/build.py first.", file=sys.stderr)
     raise
@@ -243,3 +243,10 @@ def test_copy_multithreading():
 
     h2.update(b3, multithreading=True)
     assert h2.digest() == h3.digest(), "Update state of copy diverged from expected state"
+
+
+def test_version():
+    # Just sanity check that it's a version string. No need to try to go out of
+    # our way to parse Cargo.toml.
+    assert type(__version__) is str
+    assert len(__version__.split(".")) == 3