def test_density_knotlocs(self, adata_cflare: AnnData):
g = GAMR(adata_cflare, knotlocs="density")
g.prepare(adata_cflare.var_names[0], "0", n_test_points=300).fit()
g.predict(level=0.95)
assert g.y_test.shape == (300,)
assert g.conf_int.shape == (300, 2)
def _create_gamr_model(_adata: AnnData) -> Optional[GAMR]:
try:
m = GAMR(_adata)
m.prepare(_adata.var_names[0], "0").fit()
m.predict(level=0.95)
return m
except:
return None
def test_manually_call_conf_int_not_in_predict(self, adata_cflare: AnnData):
g = GAMR(adata_cflare).prepare(adata_cflare.var_names[0], "1").fit()
g.predict(level=None)
assert g.conf_int is None
ci_95 = g.confidence_interval(level=0.95)
np.testing.assert_array_equal(g.conf_int, ci_95)
ci_100 = g.confidence_interval(level=1)
np.testing.assert_array_equal(g.conf_int, ci_100)
assert not np.allclose(ci_95, ci_100)
