Updated 1272 to 1300

papers/MCMC/plot_pdm_pz.ipynb

@@ -338,14 +338,13 @@
     "            save=False,\n",
     "            name=survey_list[0][j].name + \"pzdm.pdf\",\n",
     "            DMlines=DMlines,\n",
-    "            # DMlims=[1500],\n",
+    "            DMlims=[1500],\n",
     "            FRBZ=Zs,\n",
     "            FRBDM=DMs,\n",
     "            FRBZ2=ignored_Zs,\n",
     "            FRBDM2=ignored_DMs,\n",
     "            data_clr='r',\n",
     "            data_clr2='tab:blue'\n",
-    "            # ylabel=r\"DM$_{\\mathrm{EG}}$ (pc cm$^{-3}$)\"\n",
     "        )"
    ]
   },
@@ -429,7 +428,7 @@
     "                plt.vlines(np.log10(bEobs), np.ones(len(bEobs)) * np.log10(np.min(pE)), np.ones(len(bEobs)) * np.log10(np.max(pE)), ls='--', colors=colours[i])\n",
     "\n",
     "    plt.tight_layout()\n",
-    "    plt.savefig(\"pE.pdf\")\n",
+    "    # plt.savefig(\"pE.pdf\")\n",
     "        "
    ]
   }

zdm/loading.py

@@ -136,7 +136,7 @@ def surveys_and_grids(init_state=None, alpha_method=1,
         survey_names = ['CRAFT/FE', 
                     'CRAFT_ICS_1632',
                     'CRAFT_ICS_892', 
-                    'CRAFT_ICS_1272',
+                    'CRAFT_ICS_1300',
                     'PKS/Mb']
     if add_20220610A:
         survey_names[3] = 'CRAFT_ICS_w_220610'

zdm/misc_functions.py

@@ -2499,6 +2499,7 @@ def plot_grid_2(
     H0=None,
     showplot=False,
     DMlines=None,
+    DMlims=None,
     markersize=10,
     clim=False,
     data_clr="red",
@@ -2767,7 +2768,12 @@ def plot_grid_2(
             zstop /= zvals[1] - zvals[0]
             DM /= dmvals[1] - dmvals[0]
             plt.plot([0, zstop], [DM, DM], color=data_clr, linestyle=":")
-
+
+    if DMlims is not None:
+        for DMlim in DMlims:
+            DMlim /= dmvals[1] - dmvals[0]
+            ax.axhline(DMlim, 0, 1, color=data_clr, linestyle="-")
+
     # performs plots for the pdmgz variable
     if pdmgz is not None:
         styles = ['-','-','-']

zdm/tests/test_cube.py

@@ -118,4 +118,4 @@ def test_cube_run():
     assert check_accuracy(zdm_v1,zdm_v2)
 
 
-#test_cube_run()
+test_cube_run()

zdm/tests/test_energetics.py

@@ -22,4 +22,4 @@ def test_init_gamma():
                       float(energetics.igamma_linear[-1](1.)),
                       rtol=1e-3)
 
-
+test_init_gamma()

zdm/tests/test_parameters.py

@@ -8,4 +8,5 @@ def test_init_state():
 
     # Fuss a bit
     assert state.analysis.NewGrids 
-
+
+test_init_state()

zdm/tests/test_pcosmic.py

@@ -24,4 +24,5 @@ def test_mean_DM():
 
     # Test
     assert np.isclose(DMs[-1], 924.81566918)
-
+
+test_mean_DM()