def test_frequency_init(self):
"""fs should be initialized from device metadata if not provided"""
device = LslDevice(connection_params={}, fs=None)
device.connect()
device.acquisition_init()
self.assertEqual(device.fs, self.hz)
def test_read_data(self):
"""Should produce a valid data record."""
device = LslDevice(connection_params={}, fs=self.hz)
device.connect()
device.acquisition_init()
data = device.read_data()
self.assertTrue(len(data) > 0)
self.assertEqual(len(data), len(device.channels))
for channel in data[0:-1]:
self.assertTrue(isinstance(channel, float))
def test_incorrect_frequency(self):
"""Provided fs should match sample rate read from device"""
device = LslDevice(connection_params={},
channels=self.channels,
fs=300)
self.assertEqual(device.fs, 300)
device.connect()
with pytest.raises(Exception):
device.acquisition_init()
def test_channel_init(self):
"""Channels should be initialized from device metadata if not
provided."""
device = LslDevice(connection_params={}, fs=self.hz, channels=[])
self.assertEqual(len(device.channels), 0)
device.connect()
device.acquisition_init()
self.assertEqual(len(device.channels), self.channel_count + 1,
"Should have added a TRG channel")
self.assertEqual(device.channels[0:-1], self.channels)
def test_mismatched_channel_names(self):
"""Provided channel names should match device information."""
channels = ['ch' + str(i) for i in range(self.channel_count)]
device = LslDevice(connection_params={}, fs=self.hz, channels=channels)
self.assertEqual(len(device.channels), self.channel_count)
device.connect()
with pytest.raises(Exception):
device.acquisition_init()
def test_incorrect_number_of_channels(self):
"""A list of channels with len that does not match channel_count should
raise an exception."""
device = LslDevice(connection_params={},
fs=self.hz,
channels=['ch1', 'ch2'])
self.assertEqual(len(device.channels), 2)
self.assertNotEqual(len(device.channels), self.channel_count)
device.connect()
with pytest.raises(Exception):
device.acquisition_init()
def test_connect(self):
"""Should require a connect call before initialization."""
device = LslDevice(connection_params={}, channels=[], fs=None)
with pytest.raises(Exception):
device.acquisition_init()