summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorROUVREAU Vincent <vincent.rouvreau@inria.fr>2020-01-15 10:20:22 +0100
committerROUVREAU Vincent <vincent.rouvreau@inria.fr>2020-01-15 10:20:22 +0100
commit95c34035645e788d98a3fa6b059ad48024596906 (patch)
tree2208e03805bcceab9e744fc26deafab79030ce04
parent8c20dd2dc4b3b192badb6d26c27efdaaa32cd9bd (diff)
In python2 itertools zip_longest is named izip_longest
-rwxr-xr-xsrc/python/test/test_alpha_complex.py9
1 files changed, 7 insertions, 2 deletions
diff --git a/src/python/test/test_alpha_complex.py b/src/python/test/test_alpha_complex.py
index 712a50b6..0d9e9e45 100755
--- a/src/python/test/test_alpha_complex.py
+++ b/src/python/test/test_alpha_complex.py
@@ -11,8 +11,13 @@
from gudhi import AlphaComplex, SimplexTree
import math
import numpy as np
-import itertools
import pytest
+try:
+ # python3
+ from itertools import zip_longest
+except ImportError:
+ # python2
+ from itertools import izip_longest as zip_longest
__author__ = "Vincent Rouvreau"
__copyright__ = "Copyright (C) 2016 Inria"
@@ -114,6 +119,6 @@ def test_safe_alpha_persistence_comparison():
diag1 = simplex_tree1.persistence()
diag2 = simplex_tree2.persistence()
- for (first_p, second_p) in itertools.zip_longest(diag1, diag2):
+ for (first_p, second_p) in zip_longest(diag1, diag2):
assert first_p[0] == pytest.approx(second_p[0])
assert first_p[1] == pytest.approx(second_p[1])