Unit test for multiple TOML

See #19
CMB-S4 · Dec 2, 2020 · cbc9921 · cbc9921
1 parent 0c69569
commit cbc9921
Showing 1 changed file with 60 additions and 1 deletion.
diff --git a/05_cli.ipynb b/05_cli.ipynb
@@ -493,12 +493,71 @@
     "hp.mollview(output_map[2], min=-1e-5, max=1e-5, unit=\"K\", title=\"Total U\")"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Test multiple TOML files "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%%file only_CMB_scalar.toml\n",
+    "[sky_emission]\n",
+    "foreground_emission = 0\n",
+    "CMB_unlensed = 1\n",
+    "CMB_lensing_signal = 1\n",
+    "CMB_tensor_to_scalar_ratio = 0"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "sim2 = S4RefSimTool([\"s4_reference_design.toml\", \"only_CMB_scalar.toml\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "assert sim2.config[\"sky_emission\"][\"foreground_emission\"] == 0"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "assert sim2.config[\"sky_emission\"][\"CMB_tensor_to_scalar_ratio\"] == 0"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "assert sim.config[\"telescopes\"] == sim2.config[\"telescopes\"]"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": []
+   "source": [
+    "!rm only_CMB_scalar.toml"
+   ]
   }
  ],
  "metadata": {