def test_precomputed_no_adata(self):
pk = PrecomputedKernel(random_transition_matrix(50))
pk.write_to_adata()
assert isinstance(pk.adata, AnnData)
assert pk.adata.shape == (50, 1)
assert pk.adata.obs.shape == (50, 0)
assert pk.adata.var.shape == (1, 0)
assert "T_fwd_params" in pk.adata.uns.keys()
np.testing.assert_array_equal(pk.adata.obsp["T_fwd"].toarray(),
pk.transition_matrix.toarray())
def test_precomputed_from_kernel(self, adata: AnnData):
vk = VelocityKernel(adata).compute_transition_matrix(
mode="stochastic",
softmax_scale=4,
)
pk = PrecomputedKernel(vk)
pk.write_to_adata()
assert pk.adata is vk.adata
assert pk._origin == str(vk).strip("~<>")
assert pk.params is not vk.params
assert pk.params == vk.params
assert pk.transition_matrix is not vk.transition_matrix
np.testing.assert_array_equal(pk.transition_matrix.A,
vk.transition_matrix.A)