diff --git a/src/lib.rs b/src/lib.rs index 19fb7f1..dfa2808 100644 --- 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 index 1a60694..273ec60 100644 --- 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