def test_dataset_label(self): test_ds = DataSet('test_ds_1') test_ds.save() self.assertEqual(test_ds.label, 'test_ds_1') if pyfusion.orm_manager.IS_ACTIVE: session = pyfusion.orm_manager.Session() db_ods = session.query(DataSet).filter_by(label='test_ds_1')
def test_ORM_flucstrucs(self): """ check that flucstrucs can be saved to database""" n_ch = 10 n_samples = 1024 multichannel_data = get_multimode_test_data( channels=get_n_channels(n_ch), timebase=Timebase(np.arange(n_samples) * 1.e-6), noise=0.01) # produce a dataset of flucstrucs #print ">> ", multichannel_data.channels fs_data = multichannel_data.flucstruc(min_dphase=-2 * np.pi) print type(fs_data) #print list(fs_data)[0].dphase[0].channel_1 #print '---' # save our dataset to the database fs_data.save() if pyfusion.orm_manager.IS_ACTIVE: session = pyfusion.orm_manager.Session() d1 = DataSet('test_dataset_1') d1.save() d2 = DataSet('test_dataset_2') d2.save() # get our dataset from database our_dataset = session.query(DataSet).order_by("id").first() self.assertEqual(our_dataset.created, fs_data.created) self.assertEqual(len([i for i in our_dataset.data]), len(our_dataset)) #check flucstrucs have freq, t0 and d_phase.. #for i in our_dataset.data: # print i #print 'w' #assert False #our guinea pig flucstruc: test_fs = our_dataset.pop() self.assertTrue(isinstance(test_fs.freq, float)) self.assertTrue(isinstance(test_fs.t0, float)) # now, are the phase data correct? self.assertTrue(isinstance(test_fs.dphase, BaseOrderedDataSet)) self.assertEqual(len(test_fs.dphase), n_ch - 1) # what if we close the session and try again? session.close() session = pyfusion.orm_manager.Session() ds_again = session.query(DataSet).order_by("id").first() fs_again = list(ds_again)[0] """
def test_ORM_flucstrucs(self): """ check that flucstrucs can be saved to database""" n_ch = 10 n_samples = 1024 multichannel_data = get_multimode_test_data(channels=get_n_channels(n_ch), timebase = Timebase(np.arange(n_samples)*1.e-6), noise = 0.01) # produce a dataset of flucstrucs #print ">> ", multichannel_data.channels fs_data = multichannel_data.flucstruc(min_dphase = -2*np.pi) print type(fs_data) #print list(fs_data)[0].dphase[0].channel_1 #print '---' # save our dataset to the database fs_data.save() if pyfusion.orm_manager.IS_ACTIVE: session = pyfusion.orm_manager.Session() d1 = DataSet('test_dataset_1') d1.save() d2 = DataSet('test_dataset_2') d2.save() # get our dataset from database our_dataset = session.query(DataSet).order_by("id").first() self.assertEqual(our_dataset.created, fs_data.created) self.assertEqual(len([i for i in our_dataset.data]), len(our_dataset)) #check flucstrucs have freq, t0 and d_phase.. #for i in our_dataset.data: # print i #print 'w' #assert False #our guinea pig flucstruc: test_fs = our_dataset.pop() self.assertTrue(isinstance(test_fs.freq, float)) self.assertTrue(isinstance(test_fs.t0, float)) # now, are the phase data correct? self.assertTrue(isinstance(test_fs.dphase, BaseOrderedDataSet)) self.assertEqual(len(test_fs.dphase), n_ch-1) # what if we close the session and try again? session.close() session = pyfusion.orm_manager.Session() ds_again = session.query(DataSet).order_by("id").first() fs_again = list(ds_again)[0] """