Method check_labels

graphlearn/python/tests/test_node.py:57–61

(self, nodes)

    self.assertAlmostEqual(0.8, nodes.weights[2])

  def check_labels(self, nodes):
    self.assertEqual(3, nodes.labels.shape[0])
    self.assertEqual(2, nodes.labels[0])
    self.assertEqual(5, nodes.labels[1])
    self.assertEqual(8, nodes.labels[2])

  def check_attrs(self, nodes):
    self.assertListEqual([3, 2], list(nodes.int_attrs.shape))  # [int_num, batch_size]