class Limbo_internalPutGet(StoreDependentTestCase): def setUp(self): super(Limbo_internalPutGet, self).setUp() self.limbo = Limbo() def test_put(self): id = self.limbo.random_ObjectID(1) self.limbo._put(1, id[0]) self.assertEqual(1, self.limbo.client.get(id[0])) def test_get(self): id = self.limbo.put(1, 'one') self.limbo.updateStored('one', id) self.assertEqual(self.limbo._get('one'), 1) def test__getNonexistent(self): # Handle exception thrown with self.assertRaises(Exception) as cm: # Check that the exception thrown is a ObjectNotFoundError self.limbo._get('three') self.assertEqual(cm.exception.name, 'ObjectNotFoundError') self.assertEqual(cm.exception.message, 'Cannnot find object with ID/name "three"') def tearDown(self): super(Limbo_internalPutGet, self).tearDown()
class Limbo_UpdateStored(StoreDependentTestCase): def setUp(self): super(Limbo_UpdateStored, self).setUp() self.limbo = Limbo() # Accessing self.limbo.stored directly to test getStored separately def test_updateGet(self): self.limbo.put(1, 'one') self.limbo.updateStored('one', 3) self.assertEqual(3, self.limbo.stored['one']) def tearDown(self): super(Limbo_UpdateStored, self).tearDown()
class Limbo_sparseMatrix(StoreDependentTestCase): def setUp(self): super(Limbo_sparseMatrix, self).setUp() self.limbo = Limbo() def test_csc(self): csc = csc_matrix((3, 4), dtype=np.int8) self.limbo.put(csc, "csc") self.assertTrue( np.allclose(self.limbo.get("csc").toarray(), csc.toarray())) def tearDown(self): super(Limbo_sparseMatrix, self).tearDown()
class Limbo_getListandAll(StoreDependentTestCase): def setUp(self): super(Limbo_getListandAll, self).setUp() self.limbo = Limbo() def getListandAll(self): id = self.limbo.put(1, 'one') id2 = self.limbo.put(2, 'two') id3 = self.limbo.put(3, 'three') self.assertEqual([1, 2], self.limbo.getList(['one', 'two'])) self.assertEqual([1, 2, 3], self.limbo.get_all()) def tearDown(self): super(Limbo_getListandAll, self).tearDown()
class Limbo_GetStored(StoreDependentTestCase): def setUp(self): super(Limbo_GetStored, self).setUp() self.limbo = Limbo() def test_getStoredEmpty(self): self.assertFalse(self.limbo.getStored()) def test_putGetStored(self): self.limbo.put(1, 'one') self.assertEqual(1, self.limbo.getID( self.limbo.getStored()['one'])) # returns ID def tearDown(self): super(Limbo_GetStored, self).tearDown()
class Acquirer_Setup(StoreDependentTestCase): ''' Test creation of FileAcquirer object and store connection along with various components ''' def setUp(self): super(Acquirer_Setup, self).setUp() self.limbo = Limbo() self.acq = FileAcquirer('acq', self.limbo) self.acq.setupAcquirer('/Users/hawkwings/Documents/Neuro/RASP/rasp/data/Tolias_mesoscope_1.hdf5') # def test_init(self): # #self.acq.setupAcquirer('/Users/hawkwings/Documents/Neuro/RASP/rasp/data/zf1.h5') # self.assertTrue(1) def test_frame_acq(self): #self.acq.setupAcquirer('/Users/hawkwings/Documents/Neuro/RASP/rasp/data/zf1.h5') frame = self.acq.getFrame(1) print(frame.shape) self.assertTrue(frame.shape==(440,256)) def test_store_frame(self): for _ in range(1010): self.acq.runAcquirer() frame = self.limbo.get('curr_frame') self.assertTrue(frame.shape==(440,256)) def tearDown(self): super(Acquirer_Setup, self).tearDown()
class Limbo_saveTweak(StoreDependentTestCase): def setUp(self): super(Limbo_saveTweak, self).setUp() self.limbo = Limbo() def test_tweak(self): fileName = 'data/tweak_dump' id = self.limbo.put(1, 'one') id2 = self.limbo.put(2, 'two') tweak_ids = [id, id2] self.limbo.saveTweak(tweak_ids) with open(fileName, 'rb') as output: self.assertEqual(pickle.load(output), [1, 2]) def tearDown(self): super(Limbo_saveTweak, self).tearDown()
class Limbo_Put(StoreDependentTestCase): def setUp(self): super(Limbo_Put, self).setUp() self.limbo = Limbo() def test_putOne(self): id = self.limbo.put(1, 'one') self.assertEqual(1, self.limbo.get('one')) def test_put_twice(self): id = self.limbo.put(2, 'two') id2 = self.limbo.put(2, 'two') self.assertRaises(PlasmaObjectExists) def tearDown(self): super(Limbo_Put, self).tearDown()
class Limbo_Get(StoreDependentTestCase): def setUp(self): super(Limbo_Get, self).setUp() self.limbo = Limbo() def test_init_empty(self): self.assertFalse(self.limbo.get_all()) def tearDown(self): super(Limbo_Get, self).tearDown()
class Limbo_ReleaseReset(StoreDependentTestCase): def setUp(self): super(Limbo_ReleaseReset, self).setUp() self.limbo = Limbo() def test_release(self): self.limbo.release() self.limbo.put(1, 'one') self.assertRaises(ArrowIOError) def test_reset(self): self.limbo.reset() self.limbo.put(1, 'one') self.assertEqual(self.limbo.get('one'), 1) def tearDown(self): super(Limbo_ReleaseReset, self).tearDown()
class Limbo_Connect(StoreDependentTestCase): def setUp(self): super(Limbo_Connect, self).setUp() self.limbo = Limbo() def test_Connect(self): store_loc = '/tmp/store' self.limbo.connectStore(store_loc) self.assertIsInstance(self.limbo.client, plasma.PlasmaClient) def test_failToConnect(self): store_loc = 'asdf' # Handle exception thrown with self.assertRaises(Exception) as cm: self.limbo.connectStore(store_loc) # Check that the exception thrown is a CannotConnectToStoreError self.assertEqual(cm.exception.name, 'CannotConnectToStoreError') def tearDown(self): super(Limbo_Connect, self).tearDown()
class Limbo_PutGet(StoreDependentTestCase): def setUp(self): super(Limbo_PutGet, self).setUp() self.limbo = Limbo() def test_getOne(self): id = self.limbo.put(1, 'one') id2 = self.limbo.put(2, 'two') self.assertEqual(1, self.limbo.get('one')) self.assertEqual(id, self.limbo.stored['one']) def test_get_nonexistent(self): # Handle exception thrown with self.assertRaises(Exception) as cm: self.limbo.get('three') # Check that the exception thrown is a CannotGetObjectError self.assertEqual(cm.exception.name, 'CannotGetObjectError') def tearDown(self): super(Limbo_PutGet, self).tearDown()
class Limbo_GetID(StoreDependentTestCase): #Check both hdd_only=False/True #Check isInstance type, isInstance bytes, else def setUp(self): super(Limbo_GetID, self).setUp() self.limbo = Limbo() def test_isMatrix(self): #also tests put matrix mat = csc_matrix((3, 4), dtype=np.int8) x = self.limbo.put(mat, 'matrix') #returns object_id self.assertIsInstance(self.limbo.getID(x), csc_matrix) def test_notPut(self): obj = self.limbo.getID(self.limbo.random_ObjectID(1)) self.assertRaises(ObjectNotFoundError) def UseHDD(self): self.lmdb_store.put(1, 'one') assertEqual(self.lmdb_store.getID('one', hdd_only=True), 1) def tearDown(self): super(Limbo_GetID, self).tearDown()
class Caiman_Setup(StoreDependentTestCase): ''' Test creation of OnACID object and store connection ''' def setUp(self): super(Caiman_Setup, self).setUp() self.limbo = Limbo() self.proc = cp() self.proc.setStore(self.limbo) def test_StartProc(self): #cp.setupProcess(self.proc, 'params_dict') self.assertTrue(1) def test_runProc(self): self.proc.setup() fnames = self.limbo.get('params_dict')['fnames'] print('Test runProc: processing files: ', fnames) t = time.time() self.proc.runProcess() print('time is ', time.time() - t) self.assertTrue(1) def tearDown(self): super(Caiman_Setup, self).tearDown()
def setUp(self): super(Limbo_UpdateStored, self).setUp() self.limbo = Limbo()
def setUp(self): super(Acquirer_Setup, self).setUp() self.limbo = Limbo() self.acq = FileAcquirer('acq', self.limbo) self.acq.setupAcquirer('/Users/hawkwings/Documents/Neuro/RASP/rasp/data/Tolias_mesoscope_1.hdf5')
def setUp(self): super(Actor_setStore, self).setUp() self.actor = Actor('test') self.limbo = Limbo()
def setUp(self): super(Caiman_Setup, self).setUp() self.limbo = Limbo() self.proc = cp() self.proc.setStore(self.limbo)
def setUp(self): super(Limbo_GetStored, self).setUp() self.limbo = Limbo()
def setUp(self): super(Limbo_ReleaseReset, self).setUp() self.limbo = Limbo()
def setUp(self): super(Limbo_PutGet, self).setUp() self.limbo = Limbo()
def setUp(self): super(Limbo_getListandAll, self).setUp() self.limbo = Limbo()
def setUp(self): super(Limbo_GetID, self).setUp() self.limbo = Limbo()
def setUp(self): super(Limbo_Connect, self).setUp() self.limbo = Limbo()
def setUp(self): super(Limbo_internalPutGet, self).setUp() self.limbo = Limbo()
def setUp(self): super(Limbo_sparseMatrix, self).setUp() self.limbo = Limbo()
def setUp(self): super(Limbo_saveTweak, self).setUp() self.limbo = Limbo()