diff --git a/test/test_spatio_temporal_dataset/test_transformation.py b/test/test_spatio_temporal_dataset/test_transformation.py index e03b0eb5d1dbd99c70367432b33a7ebc26581f65..968a48c21e25130241639d350113394fa8204f79 100644 --- a/test/test_spatio_temporal_dataset/test_transformation.py +++ b/test/test_spatio_temporal_dataset/test_transformation.py @@ -2,18 +2,23 @@ import unittest import numpy as np +from spatio_temporal_dataset.coordinates.spatial_coordinates.coordinates_1D import LinSpaceSpatialCoordinates from spatio_temporal_dataset.coordinates.spatio_temporal_coordinates.generated_spatio_temporal_coordinates import \ GeneratedSpatioTemporalCoordinates, UniformSpatioTemporalCoordinates from spatio_temporal_dataset.coordinates.temporal_coordinates.generated_temporal_coordinates import \ ConsecutiveTemporalCoordinates from spatio_temporal_dataset.coordinates.transformed_coordinates.transformation.abstract_transformation import \ - CenteredScaledNormalization + CenteredScaledNormalization, IdentityTransformation from spatio_temporal_dataset.coordinates.transformed_coordinates.transformation.uniform_normalization import \ BetweenZeroAndOneNormalization, BetweenMinusOneAndOneNormalization class TestTransformation(unittest.TestCase): + def test_default_transformation(self): + coordinates = LinSpaceSpatialCoordinates.from_nb_points(3) + self.assertIsInstance(coordinates.transformation, IdentityTransformation) + def test_temporal_normalization(self): nb_steps = 3 start = 1950