update test for py3

This commit is contained in:
Vighnesh Birodkar
2014-06-26 00:57:50 +05:30
parent cbfa8aa6e1
commit b1284ee180
+6 -1
View File
@@ -14,7 +14,12 @@ def _max_edge(g, src, dst, neighbor):
except KeyError:
w2 = None
return max(w1, w2)
if w1 == None:
return w2
elif w2 == None:
return w1
else:
return max(w1, w2)
def test_rag_merge():