add test linear mapping class

Author: rflamary

Makefile

@@ -56,6 +56,11 @@ rdoc :
 
 notebook :
 	ipython notebook --matplotlib=inline  --notebook-dir=notebooks/
+	
+autopep8 :
+	autopep8 -ir test ot examples
 
+aautopep8 :
+	autopep8 -air test ot examples
 
 FORCE :

test/test_da.py

@@ -462,6 +462,30 @@ def test_linear_mapping():
     np.testing.assert_allclose(Ct, Cst, rtol=1e-2, atol=1e-2)
 
 
+def test_linear_mapping_class():
+
+    ns = 150
+    nt = 200
+
+    Xs, ys = get_data_classif('3gauss', ns)
+    Xt, yt = get_data_classif('3gauss2', nt)
+
+    otmap = ot.da.LinearTransport()
+
+    otmap.fit(Xs=Xs, Xt=Xt)
+    assert hasattr(otmap, "A_")
+    assert hasattr(otmap, "B_")
+    assert hasattr(otmap, "A1_")
+    assert hasattr(otmap, "B1_")
+
+    Xst = otmap.transform(Xs=Xs)
+
+    Ct = np.cov(Xt.T)
+    Cst = np.cov(Xst.T)
+
+    np.testing.assert_allclose(Ct, Cst, rtol=1e-2, atol=1e-2)
+
+
 def test_otda():
 
     n_samples = 150  # nb samples