add test for save_image

js/widget.js

@@ -31,6 +31,10 @@ function render({ model, el }) {
         // loop through the data and update the model
         // console.log("viewerUpdated: ", data);
         for (const key in data) {
+            // skip atomScales, modelSticks, modelPolyhedras
+            if (key === "atomScales" || key === "modelSticks" || key === "modelPolyhedras") {
+                continue;
+            }
             model.set(key, data[key]);
         }
         model.save_changes();

tests/notebooks/tests/notebooks/widgets.ipynb

@@ -252,11 +252,31 @@
     "viewer.avr.atom_scales = [1 if atom.symbol == \"Pt\" else 0.4 for atom in slab]\n",
     "viewer"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "## save image\n",
+    "# get path of current file\n",
+    "import os\n",
+    "from ipywidgets import Image\n",
+    "import time\n",
+    "\n",
+    "path = os.getcwd()\n",
+    "viewer.save_image(os.path.join(path, \"test.png\"))\n",
+    "\n",
+    "time.sleep(5)\n",
+    "\n",
+    "Image(value=open(os.path.join(path, \"test.png\"), \"rb\").read())"
+   ]
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "aiida",
    "language": "python",
    "name": "python3"
   },