def test_write_adata_key(self, adata: AnnData): vk = VelocityKernel(adata).compute_transition_matrix(softmax_scale=4) vk.write_to_adata(key="foo") assert adata is vk.adata assert "foo_params" in adata.uns.keys() np.testing.assert_array_equal(adata.obsp["foo"].toarray(), vk.transition_matrix.toarray())
def test_precomputed_adata_origin(self, adata: AnnData): vk = VelocityKernel(adata).compute_transition_matrix(mode="stochastic", softmax_scale=4) vk.write_to_adata("foo") pk = PrecomputedKernel("foo", adata=adata) assert pk._origin == "adata.obsp['foo']"