Update SmokeTest.py

Updated for new tests

Author: dsblank

panels/SmokeTest.py

@@ -1,28 +1,43 @@
-from comet_ml import API, ui
+%pip install aitk
+
 import os
+from comet_ml import API
+
+st.markdown("## Smoke Tests")
+
+print("You are running compute engine %s" % os.environ["ENGINE_VERSION"])
+
+
+st.markdown("### 1. Using `%pip magic` to install additional packages?")
+print(":white_check_mark: Pass! `%pip` works")
+
+st.markdown("### 2. Import additional packages?")
 
-print("1. Trying **to import** additional packages...")
 try:
-    import st_audio_spectrogram
-    print(":white_check_mark: Pass!")
-except ImportError:
+    import aitk
+    print(":white_check_mark: Pass! Can `import` %pip-installed packages")
+except Exception:
     print(":x: Failed! Not a current compute-engine image.")
-    
-print("2. Trying **to install** and **import** additional packages...")
-os.system("pip install snowflake-connector-python")
+
+st.markdown("### 3. Import pre-installed packages?")
 try:
-    import snowflake.connector
+    import st_audio_spectrogram
     print(":white_check_mark: Pass!")
 except ImportError:
-    print(":x: Failed! Not a current compute-engine image.")
+    print(":x: Failed! Pre-installed packages not found.")
+
 
-print("2. Test number of experiments in this project...")
+print("### 4. Test number of experiments in this project?")
 api = API()
 count = api.get_panel_experiments()
-print(f"There are {len(count)} experiments in this project. Is this correct?")
+print(f"There should be {len(count)} experiments selected in this project. Is this correct?")
 
-print("3. You should see a nice message below, not a stack trace:")
+print("### 5. Test parallel imports")
 
-api = API("645645")
-api.get_workspaces()
+print("Add two copies of this Smoke Test panel to this view," +
+      " and save the view.")
+print("Press the **Restart Session** button below and refresh your browser.")
+print("The two panels should load, first one, then the other.")
 
+if st.button("Restart Session"):
+    os.system("pkill -9 python")