modified parametric test to use rate switching

Becksteinlab · Jan 5, 2024 · 352eb72 · 352eb72
1 parent 0d5f4ef
commit 352eb72
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/basicrta/tests/test_parametric.py b/basicrta/tests/test_parametric.py
@@ -24,13 +24,13 @@ def test_parametric():
         G.results.mcweights[i] = G.results.mcweights[i][sortinds] 
         G.results.mcrates[i] = G.results.mcrates[i][sortinds]
 
-    tmp = np.array([np.sort(G.results.weights[:,i]) for i in range(G.results.ncomp)])
+    tmp = np.array([np.sort(G.results.rates[:,i]) for i in range(G.results.ncomp)])
     tmp2 = (tmp.cumsum(axis=1).T/tmp.cumsum(axis=1).T[-1])
     tmp3 = tmp.T[[np.where((tmp2[:,i]>0.025)&(tmp2[:,i]<0.975))[0] for i in range(G.results.ncomp)][0]]
-    descsort = np.median(G.results.mcweights, axis=0).argsort()[::-1]
+    descsort = np.median(G.results.mcrates, axis=0).argsort()[::-1]
     ci = np.array([[line[0],line[-1]] for line in tmp3.T])
 
-    Bools = np.array([(wts[i]>ci[descsort][i,0])&(wts[i]<ci[descsort][i,1]) for i in descsort])
+    Bools = np.array([(rts[i]>ci[descsort][i,0])&(rts[i]<ci[descsort][i,1]) for i in descsort])
 
     assert Bools.all() == True