From c6b874f53a54dc0f4464632af97f0551ac97fd13 Mon Sep 17 00:00:00 2001
From: Le Roux Erwan <erwan.le-roux@irstea.fr>
Date: Mon, 13 May 2019 14:24:30 +0200
Subject: [PATCH] [COORDINATES] improve test_transformation
---
.../test_transformation.py | 11 +++++++++--
1 file changed, 9 insertions(+), 2 deletions(-)
diff --git a/test/test_spatio_temporal_dataset/test_transformation.py b/test/test_spatio_temporal_dataset/test_transformation.py
index c6f2342c..96ed89ca 100644
--- a/test/test_spatio_temporal_dataset/test_transformation.py
+++ b/test/test_spatio_temporal_dataset/test_transformation.py
@@ -35,13 +35,20 @@ class TestTransformation(unittest.TestCase):
for transformation_class, expected in transformation_class_to_expected.items():
coordinates = UniformSpatioTemporalCoordinates.from_nb_points_and_nb_steps(nb_points=2, nb_steps=50,
transformation_class=transformation_class)
- temporal_coordinates = coordinates.temporal_coordinates
- normalized_coordinates = temporal_coordinates.df_coordinates().iloc[:, 0].values
+ # Temporal coordinates, the order is known
+ normalized_coordinates = coordinates.temporal_coordinates.df_coordinates().iloc[:, 0].values
normalized_coordinates = np.array([normalized_coordinates[0], normalized_coordinates[-1]])
expected_coordinates = np.array(expected)
equals = normalized_coordinates == expected_coordinates
self.assertTrue(equals.all(),
msg="expected: {}, res:{}".format(expected_coordinates, normalized_coordinates))
+ # Spatial coordinates, we do not know the order
+ normalized_coordinates = coordinates.temporal_coordinates.df_coordinates().iloc[:, 0].values
+ normalized_coordinates = {normalized_coordinates[0], normalized_coordinates[-1]}
+ expected_coordinates = set(expected)
+ equals = normalized_coordinates == expected_coordinates
+ self.assertTrue(equals,
+ msg="expected: {}, res:{}".format(expected_coordinates, normalized_coordinates))
if __name__ == '__main__':
--
GitLab