Skip to content

Commit

Permalink
add test for estimation
Browse files Browse the repository at this point in the history
  • Loading branch information
alexandrebarachant committed May 23, 2015
1 parent 584383a commit be319fc
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions tests/test_estimation.py
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit be319fc

Please sign in to comment.