Function check_default_s_attrs

graphlearn/python/tests/utils.py:275–277

(s_attrs, count, default_string_attr="")

                          decimal=4)

def check_default_s_attrs(s_attrs, count, default_string_attr=""):
  npt.assert_equal(s_attrs.flatten(),
                   np.array([default_string_attr] * count))

def check_edge_shape(edges, batch_size):
  npt.assert_equal(edges.src_ids.reshape(-1).size, batch_size)