test_version

ghstack-source-id: feff68facf17df017a2cbe0586239e69ad462f54 Pull Request resolved: #59
pytorch · Aug 21, 2020 · aba69a5 · aba69a5
1 parent 92767c3
commit aba69a5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/test_csprng.py b/test/test_csprng.py
@@ -301,5 +301,10 @@ def measure(size):
         # Pessimistic check that parallel execution gives >= 1.5 performance boost
         self.assertTrue(time_for_1M/time_for_1K < 1000 / min(1.5, torch.get_num_threads()))
 
+    def test_version(self):
+        import torchcsprng.version as version
+        self.assertTrue(version.__version__)
+        self.assertTrue(version.git_version)
+
 if __name__ == '__main__':
     unittest.main()