Commit 14425224 authored by Le Roux Erwan's avatar Le Roux Erwan
Browse files

[refactor] add test_default_transformation

parent 04a73d0a
No related merge requests found
Showing with 6 additions and 1 deletion
+6 -1
...@@ -2,18 +2,23 @@ import unittest ...@@ -2,18 +2,23 @@ import unittest
import numpy as np 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 \ from spatio_temporal_dataset.coordinates.spatio_temporal_coordinates.generated_spatio_temporal_coordinates import \
GeneratedSpatioTemporalCoordinates, UniformSpatioTemporalCoordinates GeneratedSpatioTemporalCoordinates, UniformSpatioTemporalCoordinates
from spatio_temporal_dataset.coordinates.temporal_coordinates.generated_temporal_coordinates import \ from spatio_temporal_dataset.coordinates.temporal_coordinates.generated_temporal_coordinates import \
ConsecutiveTemporalCoordinates ConsecutiveTemporalCoordinates
from spatio_temporal_dataset.coordinates.transformed_coordinates.transformation.abstract_transformation import \ 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 \ from spatio_temporal_dataset.coordinates.transformed_coordinates.transformation.uniform_normalization import \
BetweenZeroAndOneNormalization, BetweenMinusOneAndOneNormalization BetweenZeroAndOneNormalization, BetweenMinusOneAndOneNormalization
class TestTransformation(unittest.TestCase): 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): def test_temporal_normalization(self):
nb_steps = 3 nb_steps = 3
start = 1950 start = 1950
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment