!1374 Fix pylint problem

Merge pull request !1374 from xiefangqi/xfq_fix_pylint
5 years ago · 3947c3d078
parent d402b94476 34236ce1f1
commit 3947c3d078
1 changed files with 1 additions and 1 deletions
--- a/tests/ut/python/dataset/test_graphdata.py
+++ b/tests/ut/python/dataset/test_graphdata.py
@ -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)