add version check test (#9)

d6846203 · Jeff Rasley · GitHub · 52c5a936 · d6846203
Unverified Commit d6846203 authored Feb 04, 2020 by Jeff Rasley Committed by GitHub Feb 04, 2020
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

tests/unit/test_config.py tests/unit/test_config.py +6 -0

No files found.
--- a/tests/unit/test_config.py
+++ b/tests/unit/test_config.py
@@ -7,3 +7,9 @@ import deepspeed

 def test_cuda():
    assert (torch.cuda.is_available())
+
+
+def test_check_version():
+    assert hasattr(deepspeed, "__git_hash__")
+    assert hasattr(deepspeed, "__git_branch__")
+    assert hasattr(deepspeed, "__version__")