Merge pull request #19 from rflamary/pytest

Pytest with 89% coverage

Fixes #19 

11 files changed, 44 insertions, 7 deletions

diff --git a/examples/plot_OTDA_2D.py b/examples/plot_OTDA_2D.py
index 1bda59c..f2108c6 100644
--- a/examples/plot_OTDA_2D.py
+++ b/examples/plot_OTDA_2D.py
@@ -6,6 +6,10 @@ OT for empirical distributions
 
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 import matplotlib.pylab as pl
 import ot
diff --git a/examples/plot_OTDA_classes.py b/examples/plot_OTDA_classes.py
index 4d3846a..53e4bae 100644
--- a/examples/plot_OTDA_classes.py
+++ b/examples/plot_OTDA_classes.py
@@ -6,6 +6,10 @@ OT for domain adaptation
 
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import matplotlib.pylab as pl
 import ot
 
diff --git a/examples/plot_OTDA_color_images.py b/examples/plot_OTDA_color_images.py
index 75ac5b6..c5ff873 100644
--- a/examples/plot_OTDA_color_images.py
+++ b/examples/plot_OTDA_color_images.py
@@ -9,6 +9,10 @@ Regularized discrete optimal transport.
 SIAM Journal on Imaging Sciences, 7(3), 1853-1882.
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 from scipy import ndimage
 import matplotlib.pylab as pl
diff --git a/examples/plot_OTDA_mapping.py b/examples/plot_OTDA_mapping.py
index a5c2b21..a0d7f8b 100644
--- a/examples/plot_OTDA_mapping.py
+++ b/examples/plot_OTDA_mapping.py
@@ -9,6 +9,10 @@ OT mapping estimation for domain adaptation [8]
     Neural Information Processing Systems (NIPS), 2016.
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 import matplotlib.pylab as pl
 import ot
diff --git a/examples/plot_OTDA_mapping_color_images.py b/examples/plot_OTDA_mapping_color_images.py
index 9710461..8064b25 100644
--- a/examples/plot_OTDA_mapping_color_images.py
+++ b/examples/plot_OTDA_mapping_color_images.py
@@ -11,6 +11,10 @@ OT for domain adaptation with image color adaptation [6] with mapping estimation
 
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 from scipy import ndimage
 import matplotlib.pylab as pl
diff --git a/examples/plot_OT_1D.py b/examples/plot_OT_1D.py
index 2f3b924..0f3a26a 100644
--- a/examples/plot_OT_1D.py
+++ b/examples/plot_OT_1D.py
@@ -4,9 +4,12 @@
 1D optimal transport
 ====================
 
-@author: rflamary
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 import matplotlib.pylab as pl
 import ot
diff --git a/examples/plot_OT_2D_samples.py b/examples/plot_OT_2D_samples.py
index 75ed7db..023e645 100644
--- a/examples/plot_OT_2D_samples.py
+++ b/examples/plot_OT_2D_samples.py
@@ -4,9 +4,12 @@
 2D Optimal transport between empirical distributions
 ====================================================
 
-@author: rflamary
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 import matplotlib.pylab as pl
 import ot
diff --git a/examples/plot_OT_L1_vs_L2.py b/examples/plot_OT_L1_vs_L2.py
index 86d902b..dfc9462 100644
--- a/examples/plot_OT_L1_vs_L2.py
+++ b/examples/plot_OT_L1_vs_L2.py
@@ -8,9 +8,12 @@ Stole the figure idea from Fig. 1 and 2 in
 https://arxiv.org/pdf/1706.07650.pdf
 
 
-@author: rflamary
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 import matplotlib.pylab as pl
 import ot
diff --git a/examples/plot_WDA.py b/examples/plot_WDA.py
index 9eb8693..42789f2 100644
--- a/examples/plot_WDA.py
+++ b/examples/plot_WDA.py
@@ -4,9 +4,12 @@
 Wasserstein Discriminant Analysis
 =================================
 
-@author: rflamary
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 import matplotlib.pylab as pl
 
diff --git a/examples/plot_barycenter_1D.py b/examples/plot_barycenter_1D.py
index ab236e1..875f44c 100644
--- a/examples/plot_barycenter_1D.py
+++ b/examples/plot_barycenter_1D.py
@@ -4,10 +4,12 @@
 1D Wasserstein barycenter demo
 ==============================
 
-
-@author: rflamary
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 import matplotlib.pylab as pl
 import ot
diff --git a/examples/plot_compute_emd.py b/examples/plot_compute_emd.py
index 558facb..893eecf 100644
--- a/examples/plot_compute_emd.py
+++ b/examples/plot_compute_emd.py
@@ -4,9 +4,12 @@
 1D optimal transport
 ====================
 
-@author: rflamary
 """
 
+# Author: Remi Flamary <remi.flamary@unice.fr>
+#
+# License: MIT License
+
 import numpy as np
 import matplotlib.pylab as pl
 import ot