p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix pylint

This commit is contained in:
xiefangqi 2020-05-22 20:18:21 +08:00
parent d402b94476
commit 34236ce1f1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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