def test_init_noindex_columns(self): from span.tdt.spikeglobals import ChannelIndex as columns spikes = SpikeDataFrame(self.raw, self.meta, columns=columns) self.assertRaises(AssertionError, SpikeDataFrame, self.raw, None) self.assertRaises(TypeError, SpikeDataFrame, self.raw) assert_array_equal(spikes.values, self.raw) assert_array_equal(spikes.columns.values, columns.values)
def test_ndtuples_1(self): n = randint(1, 3) x = ndtuples(n) assert_array_equal(x, np.arange(n))
def test_fs(self): fs = self.spikes.fs self.assertEqual(fs, self.meta.fs.max()) assert_array_equal([fs] * self.meta.fs.size, self.meta.fs) self.assertIsInstance(fs, float)
def test_init_with_index_nocolumns(self): spikes = SpikeDataFrame(self.raw, self.meta, index=self.spikes.index) assert_array_equal(spikes.values, self.raw) assert_array_equal(spikes.index.values, self.spikes.index.values)
def test_init_noindex_nocolumns(self): spikes = SpikeDataFrame(self.raw, self.meta) self.assertRaises(AssertionError, SpikeDataFrame, self.raw, None) self.assertRaises(TypeError, SpikeDataFrame, self.raw) assert_array_equal(spikes.values, self.raw)