diff --git a/test/extreme_estimator/__init__.py b/extreme_estimator/R_fit/__init__.py
similarity index 100%
rename from test/extreme_estimator/__init__.py
rename to extreme_estimator/R_fit/__init__.py
diff --git a/test/spatio_temporal_dataset/__init__.py b/extreme_estimator/R_fit/gev_fit/__init__.py
similarity index 100%
rename from test/spatio_temporal_dataset/__init__.py
rename to extreme_estimator/R_fit/gev_fit/__init__.py
diff --git a/extreme_estimator/R_fit/max_stable_fit/__init__.py b/extreme_estimator/R_fit/max_stable_fit/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/extreme_estimator/__init__.py b/extreme_estimator/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/extreme_estimator/estimator/full_estimator.py b/extreme_estimator/estimator/full_estimator.py
index 907ba60f6b726ef73579a26e5f40ef8a74884829..b5f226f112d79fb178c1133df0913fcc0aff9045 100644
--- a/extreme_estimator/estimator/full_estimator.py
+++ b/extreme_estimator/estimator/full_estimator.py
@@ -23,10 +23,8 @@ class SmoothMarginalsThenUnitaryMsp(AbstractFullEstimator):
# Estimate the margin parameters
self.margin_estimator.fit()
# Compute the maxima_frech
- print(self.dataset.maxima_gev)
maxima_frech = self.margin_estimator.margin_model.gev2frech(maxima_gev=self.dataset.maxima_gev,
df_gev_params=self.margin_estimator.df_gev_params)
- print(maxima_frech)
# Update maxima frech field through the dataset object
self.dataset.maxima_frech = maxima_frech
# Estimate the max stable parameters
diff --git a/spatio_temporal_dataset/spatial_coordinates/alps_station_2D_coordinates.py b/spatio_temporal_dataset/spatial_coordinates/alps_station_2D_coordinates.py
index aa88ff305b66d7ede56cc1c2e175c772b45e5488..a85efb226bb4bad6791c13aba11695085209f2d3 100644
--- a/spatio_temporal_dataset/spatial_coordinates/alps_station_2D_coordinates.py
+++ b/spatio_temporal_dataset/spatial_coordinates/alps_station_2D_coordinates.py
@@ -9,8 +9,8 @@ class AlpsStation2DCoordinates(AlpsStation3DCoordinates):
@classmethod
def from_csv(cls, csv_file='coord-lambert2'):
# Remove the Z coordinates from df_coord
- spatial_coordinates = super().from_csv(csv_file)
- spatial_coordinates.df_coord.drop(cls.COORD_Z, axis=1, inplace=True)
+ spatial_coordinates = super().from_csv(csv_file) # type: AlpsStation3DCoordinates
+ spatial_coordinates.df_coordinates.drop(cls.COORD_Z, axis=1, inplace=True)
return spatial_coordinates
diff --git a/spatio_temporal_dataset/spatial_coordinates/transformed_coordinates.py b/spatio_temporal_dataset/spatial_coordinates/transformed_coordinates.py
index 8f38066ce46e19ec229282b8071868c1c821689d..c2f99ac74cd10a593a0618f476919b164a26ea13 100644
--- a/spatio_temporal_dataset/spatial_coordinates/transformed_coordinates.py
+++ b/spatio_temporal_dataset/spatial_coordinates/transformed_coordinates.py
@@ -7,8 +7,8 @@ class TransformedCoordinates(AbstractSpatialCoordinates):
@classmethod
def from_coordinates(cls, spatial_coordinates: AbstractSpatialCoordinates,
transformation_function: AbstractTransformation):
- df_coord_transformed = spatial_coordinates.df_coordinates.copy()
- df_coord_transformed = transformation_function.transform(df_coord=df_coord_transformed)
- return cls(df_coord=df_coord_transformed, s_split=spatial_coordinates.s_split)
+ df_coordinates_transformed = spatial_coordinates.df_coordinates.copy()
+ df_coordinates_transformed = transformation_function.transform(df_coord=df_coordinates_transformed)
+ return cls(df_coordinates=df_coordinates_transformed, s_split=spatial_coordinates.s_split)
diff --git a/test/spatio_temporal_dataset/test_temporal_observations.py b/test/spatio_temporal_dataset/test_temporal_observations.py
deleted file mode 100644
index 0ffd73059dea5780e4d9e824fe67a206d8fa1988..0000000000000000000000000000000000000000
--- a/test/spatio_temporal_dataset/test_temporal_observations.py
+++ /dev/null
@@ -1,15 +0,0 @@
-import unittest
-
-from spatio_temporal_dataset.spatial_coordinates.alps_station_2D_coordinates import \
- AlpsStation2DCoordinatesBetweenZeroAndOne
-from spatio_temporal_dataset.spatial_coordinates.alps_station_3D_coordinates import \
- AlpsStation3DCoordinatesWithAnisotropy
-from spatio_temporal_dataset.spatial_coordinates.generated_coordinates import CircleCoordinatesRadius1
-
-
-class TestTemporalObservations(unittest.TestCase):
-
- DISPLAY = False
-
-if __name__ == '__main__':
- unittest.main()
diff --git a/test/test_extreme_estimator/__init__.py b/test/test_extreme_estimator/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/test/extreme_estimator/test_full_estimators.py b/test/test_extreme_estimator/test_full_estimators.py
similarity index 75%
rename from test/extreme_estimator/test_full_estimators.py
rename to test/test_extreme_estimator/test_full_estimators.py
index 9ec15296c44efccdd75fa816d365f2a7df7d1bf8..2a9aa6b32f020567f7e360ae9adb3b452a3c09b9 100644
--- a/test/extreme_estimator/test_full_estimators.py
+++ b/test/test_extreme_estimator/test_full_estimators.py
@@ -1,12 +1,11 @@
import unittest
+from itertools import product
-from extreme_estimator.estimator.full_estimator import FullEstimatorInASingleStep, \
- FullEstimatorInASingleStepWithSmoothMarginals, SmoothMarginalsThenUnitaryMsp
-from spatio_temporal_dataset.dataset.simulation_dataset import MarginDataset, FullSimulatedDataset
+from extreme_estimator.estimator.full_estimator import SmoothMarginalsThenUnitaryMsp
+from spatio_temporal_dataset.dataset.simulation_dataset import FullSimulatedDataset
from spatio_temporal_dataset.spatial_coordinates.generated_coordinates import CircleCoordinatesRadius1
-from test.extreme_estimator.test_margin_estimators import TestMarginEstimators
-from test.extreme_estimator.test_max_stable_estimators import TestMaxStableEstimators
-from itertools import product
+from test.test_extreme_estimator.test_margin_estimators import TestMarginEstimators
+from test.test_extreme_estimator.test_max_stable_estimators import TestMaxStableEstimators
class TestFullEstimators(unittest.TestCase):
@@ -20,7 +19,6 @@ class TestFullEstimators(unittest.TestCase):
self.margin_models = TestMarginEstimators.load_margin_models()
def test_full_estimators(self):
- print(self.margin_models, self.max_stable_models)
for margin_model, max_stable_model in product(self.margin_models, self.max_stable_models):
dataset = FullSimulatedDataset.from_double_sampling(nb_obs=10, margin_model=margin_model,
spatial_coordinates=self.spatial_coord,
diff --git a/test/extreme_estimator/test_margin_estimators.py b/test/test_extreme_estimator/test_margin_estimators.py
similarity index 100%
rename from test/extreme_estimator/test_margin_estimators.py
rename to test/test_extreme_estimator/test_margin_estimators.py
diff --git a/test/extreme_estimator/test_max_stable_estimators.py b/test/test_extreme_estimator/test_max_stable_estimators.py
similarity index 100%
rename from test/extreme_estimator/test_max_stable_estimators.py
rename to test/test_extreme_estimator/test_max_stable_estimators.py
diff --git a/test/test_spatio_temporal_dataset/__init__.py b/test/test_spatio_temporal_dataset/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391
diff --git a/test/spatio_temporal_dataset/test_spatial_coordinates.py b/test/test_spatio_temporal_dataset/test_spatial_coordinates.py
similarity index 100%
rename from test/spatio_temporal_dataset/test_spatial_coordinates.py
rename to test/test_spatio_temporal_dataset/test_spatial_coordinates.py
diff --git a/test/test_spatio_temporal_dataset/test_temporal_observations.py b/test/test_spatio_temporal_dataset/test_temporal_observations.py
new file mode 100644
index 0000000000000000000000000000000000000000..3e0657dac3e44e2853d2655147b8a83da4809e60
--- /dev/null
+++ b/test/test_spatio_temporal_dataset/test_temporal_observations.py
@@ -0,0 +1,9 @@
+import unittest
+
+
+class TestTemporalObservations(unittest.TestCase):
+ DISPLAY = False
+
+
+if __name__ == '__main__':
+ unittest.main()