Add version checks to smoke tests (pytorch#375)

sailfish009 · Oct 14, 2019 · afc3799 · afc3799
1 parent baf6817
commit afc3799
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/run_tests.sh b/run_tests.sh
@@ -115,6 +115,15 @@ if [[ "$OSTYPE" == "msys" ]]; then
     fi
 fi
 
+# Test that the version number is consistent during building and testing
+if [[ "$PYTORCH_BUILD_NUMBER" -gt 1 ]]; then
+    expected_version="${PYTORCH_BUILD_VERSION}.post${PYTORCH_BUILD_NUMBER}"
+else
+    expected_version="${PYTORCH_BUILD_VERSION}"
+fi
+echo "Checking that we are testing the package that is just built"
+python -c "import torch; exit(0 if torch.__version__ == '$expected_version' else 1)"
+
 # Test that CUDA builds are setup correctly
 if [[ "$cuda_ver" != 'cpu' ]]; then
     # Test CUDA archs