minor changes to testing notebook

spacetelescope · Jan 9, 2021 · 2a9dd6d · 2a9dd6d
1 parent da8e2fc
commit 2a9dd6d
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/Jupyter Notebooks/LUVOIR/20_drifting coro floor.ipynb b/Jupyter Notebooks/LUVOIR/20_drifting coro floor.ipynb
@@ -676,7 +676,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "coro_floor_matrix_full = np.random.normal(loc=0, scale=0.1, size=(120,120)) * coro_floor_old"
+    "coro_floor_matrix_full = np.random.normal(loc=0, scale=1, size=(120,120)) * coro_floor_old * 10"
    ]
   },
   {
@@ -798,6 +798,15 @@
     "plt.colorbar()"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "assert np.allclose(matrix_pastis_drift, pmatrix_old, rtol=1e-8, atol=1e-24), 'Calculated LUVOIR small PASTIS matrix is wrong.'"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,