Python assert_array_equal Examples

Example #1

File: test_spikedataframe.py Project: cpcloud/span

    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)

Example #2

File: test_utils.py Project: cpcloud/span

 def test_ndtuples_1(self):
     n = randint(1, 3)
     x = ndtuples(n)
     assert_array_equal(x, np.arange(n))

Example #3

File: test_spikedataframe.py Project: cpcloud/span

 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)

Example #4

File: test_spikedataframe.py Project: cpcloud/span

 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)

Example #5

File: test_spikedataframe.py Project: cpcloud/span

 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)