class LabEntryTest(unittest.TestCase): def setUp(self): db = isotope_manager_factory().db self.le = LabnumberEntry(db=db) # self.le.irradiation = 'NM-251' # self.le.level = 'H' # def testNPositions(self): # n = len(self.le.irradiated_positions) # self.assertEqual(n, 12) # # def testWrite(self): # le = self.le # le.make_table() def testLoadFile(self): p = './data/irradiation_import.xls' self.le.irradiation = 'TEST' self.le._load_positions_from_file(p, dry_run=False) self.assertEqual(self.le.irradiation, 'TEST') self.assertEqual(len(self.le.levels), 2) self.assertEqual(self.le.level, 'B') self.assertEqual(len(self.le.irradiated_positions), 61) self.assertEqual(self.le.irradiated_positions[0].sample, 'test_sample1') self.assertEqual(self.le.irradiated_positions[0].material, 'test_material') self.assertEqual(self.le.irradiated_positions[0].weight, 1.1)
def setUp(self): db = isotope_manager_factory().db self.le = LabnumberEntry(db=db)