def test_write(self): b1d = buffer1d(self.dset_1d, max_fill=10) b2d = buffer2d(self.dset_2d, max_fill=10) exp1d = np.zeros(10, dtype=int) exp1d[:9] = np.arange(9) exp2d = np.zeros((10, 100), dtype=int) exp2d[:9, :9] = np.tril(np.repeat(np.arange(9), 9).reshape(9, 9).T) for i in np.arange(9): b1d.write(i) b2d.write(np.arange(i+1)) npt.assert_equal(b1d._buf, exp1d) npt.assert_equal(b2d._buf, exp2d) for i in np.arange(2): b1d.write(i) b2d.write(np.arange(i+1)) exp1d = np.zeros(10, dtype=int) exp1d[0] = 1 exp2d = np.zeros((10, 100), dtype=int) exp2d[0, 1] = 1 npt.assert_equal(b1d._buf, exp1d) npt.assert_equal(b2d._buf, exp2d)
def test_init(self): b1d = buffer1d(self.dset_1d, max_fill=10) b2d = buffer2d(self.dset_2d, max_fill=10) npt.assert_equal(b1d.dset, self.dset_1d) npt.assert_equal(b2d.dset, self.dset_2d) self.assertEqual(b1d._n, 0) self.assertEqual(b2d._n, 0) self.assertEqual(b1d._idx, 0) self.assertEqual(b2d._idx, 0) self.assertEqual(b1d._max_fill, 10) self.assertEqual(b2d._max_fill, 10) npt.assert_equal(b1d._buf, np.zeros(10, dtype=int)) npt.assert_equal(b2d._buf, np.zeros((10, 100), dtype=int))
def test_is_full(self): b1d = buffer1d(self.dset_1d, max_fill=10) b2d = buffer2d(self.dset_2d, max_fill=10) self.assertFalse(b1d.is_full()) self.assertFalse(b2d.is_full()) b1d._n = 10 b2d._n = 10 self.assertTrue(b1d.is_full()) self.assertTrue(b2d.is_full()) b1d._n = 20 b2d._n = 20 self.assertTrue(b1d.is_full()) self.assertTrue(b2d.is_full()) b1d._n = 0 b2d._n = 0
def test_destructor(self): b1d = buffer1d(self.dset_1d, max_fill=10) b2d = buffer2d(self.dset_2d, max_fill=10) for i in np.arange(9): b1d.write(i) b2d.write(np.arange(i+1)) npt.assert_equal(b1d.dset, np.zeros(100, dtype=int)) npt.assert_equal(b2d.dset, np.zeros((100, 100), dtype=int)) del b1d del b2d exp1d = np.zeros(100, dtype=int) exp1d[:9] = np.arange(9) exp2d = np.zeros((100, 100), dtype=int) exp2d[:9, :9] = np.tril(np.repeat(np.arange(9), 9).reshape(9, 9).T) npt.assert_equal(self.dset_1d, exp1d) npt.assert_equal(self.dset_2d, exp2d)