Browse Source

fix pylint

tags/v0.3.0-alpha
xiefangqi 5 years ago
parent
commit
34236ce1f1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/ut/python/dataset/test_graphdata.py

+ 1
- 1
tests/ut/python/dataset/test_graphdata.py View File

@@ -22,7 +22,7 @@ DATASET_FILE = "../data/mindrecord/testGraphData/testdata"
def test_graphdata_getfullneighbor():
g = ds.GraphData(DATASET_FILE, 2)
nodes = g.get_all_nodes(1)
assert len(nodes) is 10
assert len(nodes) == 10
nodes_list = nodes.tolist()
neighbor = g.get_all_neighbors(nodes_list, 2)
assert neighbor.shape == (10, 6)


Loading…
Cancel
Save