Function test_mean_zero_degree

tests/python/common/test_heterograph-kernel.py:402–408

(g, idtype)

@parametrize_idtype
@pytest.mark.parametrize("g", get_cases(["homo-zero-degree"]))
def test_mean_zero_degree(g, idtype):
    g = g.astype(idtype).to(F.ctx())
    g.ndata["h"] = F.ones((g.num_nodes(), 3))
    g.update_all(fn.copy_u("h", "m"), fn.mean("m", "x"))
    deg = F.asnumpy(g.in_degrees())
    v = F.tensor(np.where(deg == 0)[0])
    assert F.allclose(F.gather_row(g.ndata["x"], v), F.zeros((len(v), 3)))


if __name__ == "__main__":