import numpy as np
from numpy.testing import assert_allclose as ac
from pytest import raises, fixture, mark
from ..session import Session
from ...gui.qt import wrap_qt
from ...utils.array import _spikes_in_clusters
from ...utils.tempdir import TemporaryDirectory
from ...utils.logging import set_level
from ...io.mock import MockModel, artificial_traces
from ...io.kwik.mock import create_mock_kwik
from ...io.kwik.creator import create_kwik
# Skip these tests in "make test-quick".
pytestmark = mark.long()
#------------------------------------------------------------------------------
# Fixtures
#------------------------------------------------------------------------------
def setup():
set_level('debug')
def teardown():
set_level('info')
n_clusters = 5
#------------------------------------------------------------------------------
from pytest import mark
import numpy as np
from numpy.testing import assert_allclose as ac
from numpy.testing import assert_array_equal as ae
from ..gui import ClusterManualGUI
from ....utils.settings import _load_default_settings
from ....utils.logging import set_level
from ....utils.array import _spikes_in_clusters
from ....io.mock import MockModel
from ....io.kwik.store_items import create_store
# Skip these tests in "make test-quick".
pytestmark = mark.long()
#------------------------------------------------------------------------------
# Kwik tests
#------------------------------------------------------------------------------
def setup():
set_level('debug')
def _start_manual_clustering(config='none', shortcuts=None):
if config is 'none':
config = []
model = MockModel()
spc = model.spikes_per_cluster
