From be319fcbb42acc2cdd9435a8664e0c8ec82359ac Mon Sep 17 00:00:00 2001
From: Alexandre Barachant <alexandre.barachant@gmail.com>
Date: Sat, 23 May 2015 19:33:38 -0400
Subject: [PATCH] add test for estimation

---
 tests/test_estimation.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 tests/test_estimation.py

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