def test_potato_partial_fit_no_poslabel(get_covmats, get_labels):
n_trials, n_channels, n_classes = 6, 3, 2
covmats = get_covmats(n_trials, n_channels)
labels = get_labels(n_trials, n_classes)
pt = Potato().fit(covmats, labels)
with pytest.raises(ValueError): # no positive labels
pt.partial_fit(covmats, [0] * n_trials)
def test_potato_partial_fit_alpha(alpha, get_covmats, get_labels):
n_trials, n_channels, n_classes = 6, 3, 2
covmats = get_covmats(n_trials, n_channels)
labels = get_labels(n_trials, n_classes)
pt = Potato().fit(covmats, labels)
with pytest.raises(ValueError):
pt.partial_fit(covmats, labels, alpha=alpha)
def test_potato_partial_fit_diff_channels(get_covmats, get_labels):
n_trials, n_channels, n_classes = 6, 3, 2
covmats = get_covmats(n_trials, n_channels)
labels = get_labels(n_trials, n_classes)
pt = Potato().fit(covmats, labels)
with pytest.raises(ValueError): # unequal # of chans
pt.partial_fit(get_covmats(2, n_channels + 1))