def test_channels(self):
db = iidb.open('test.mdb', readonly=False)
data = self._make_array((5, 5, 3))
db[234] = data
db.close()
db2 = iidb.open('test.mdb', readonly=True)
np.testing.assert_array_equal(db2[234], data)
def test_put_multiple(self):
data = [
(1, self._make_array()),
(2, self._make_array()),
]
db = iidb.open('test.mdb', readonly=False)
db.putmulti(data)
db.close()
db2 = iidb.open('test.mdb', readonly=False)
np.testing.assert_array_equal(db2[1], data[0][1])
np.testing.assert_array_equal(db2[2], data[1][1])
def test_basic_put_and_get_lz4(self):
db = iidb.open('test.mdb', readonly=False, mode=1)
data = self._make_array()
db[123] = data
self.assertTrue(123 in db)
self.assertFalse(234 in db)
self.assertFalse(db.closed)
db.close()
self.assertTrue(db.closed)
db2 = iidb.open('test.mdb', readonly=True)
np.testing.assert_array_equal(db2[123], data)
def test_context_manager(self):
with iidb.open('test.mdb', readonly=False) as db:
data = self._make_array((4, 5))
db[123] = data
self.assertEqual(db.get_image_dimension(123), (4, 5))
self.assertFalse(db.closed)
self.assertTrue(db.closed)
def test_get_dimension(self):
db = iidb.open('test.mdb', readonly=False)
data = self._make_array((4, 5))
db[123] = data
self.assertEqual(db.get_image_dimension(123), (4, 5))