diff --git a/tests/test_estimation.py b/tests/test_estimation.py new file mode 100644 index 00000000..38116539 --- /dev/null +++ b/tests/test_estimation.py @@ -0,0 +1,33 @@ +import numpy as np +from pyriemann.estimation import Covariances, ERPCovariances, XdawnCovariances, CospCovariances + +def test_covariances(): + """Test fit Covariances""" + x = np.random.randn(2,3,100) + cov = Covariances() + cov.fit(x) + cov.fit_transform(x) + +def test_ERPcovariances(): + """Test fit ERPCovariances""" + x = np.random.randn(10,3,100) + labels = np.array([0,1]).repeat(5) + cov = ERPCovariances() + cov.fit_transform(x,labels) + cov = ERPCovariances(classes=[0]) + cov.fit_transform(x,labels) + + +def test_Xdawncovariances(): + """Test fit ERPCovariances""" + x = np.random.randn(10,3,100) + labels = np.array([0,1]).repeat(5) + cov = XdawnCovariances() + cov.fit_transform(x,labels) + +def test_Cospcovariances(): + """Test fit CospCovariances""" + x = np.random.randn(2,3,1000) + cov = CospCovariances() + cov.fit(x) + cov.fit_transform(x) \ No newline at end of file