def test_baseordereddataset_label(self): test_ds = BaseOrderedDataSet('test_ods_1') test_ds.save() self.assertEqual(test_ds.label, 'test_ods_1') if pyfusion.orm_manager.IS_ACTIVE: session = pyfusion.orm_manager.Session() db_ods = session.query(BaseOrderedDataSet).filter_by(label='test_ods_1')
def test_ordered_dataset_ORM(self): channel_01 = Channel('channel_01', Coords('dummy', (0,0,0))) channel_02 = Channel('channel_02', Coords('dummy', (0,0,0))) channel_03 = Channel('channel_03', Coords('dummy', (0,0,0))) channel_04 = Channel('channel_04', Coords('dummy', (0,0,0))) fd1 = FloatDelta(channel_01, channel_02, 0.45) fd2 = FloatDelta(channel_02, channel_03, 0.25) fd3 = FloatDelta(channel_03, channel_04, 0.49) #ods = OrderedDataSet(ordered_by="channel_1.name") ods = BaseOrderedDataSet('test_ods') for fd in [fd3, fd1, fd2]: ods.append(fd) ods.save() # now read out of database if pyfusion.orm_manager.IS_ACTIVE: session = pyfusion.orm_manager.Session() db_ods = session.query(BaseOrderedDataSet).first() self.assertEqual(db_ods[0].channel_1.name, 'channel_03') self.assertEqual(db_ods[1].channel_1.name, 'channel_01') self.assertEqual(db_ods[2].channel_1.name, 'channel_02')