Exemplos de VelocityKernel.write_to_adata em Python

Exemplo n.º 1

Arquivo: test_kernels.py Projeto: nkandhari/cellrank

    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())

Exemplo n.º 2

Arquivo: test_kernels.py Projeto: changwn/cellrank

    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']"