class selectAllExceptions(unittest.TestCase): def setUp(self): self.P = PdfFit() self.iset = 1 self.i = 1 def tearDown(self): del self.P def test_unassignedError1(self): """raise pdffit2.unassignedError when set does not exist""" self.assertRaises(pdffit2.unassignedError, self.P.selectAll, self.iset, 'i') def test_unassignedError2(self): """raise pdffit2.unassignedError when set does not exist""" self.P.read_struct(datafile("Ni.stru")) # fail when there is no dataset self.assertRaises(pdffit2.unassignedError, self.P.selectAll, self.iset, 'i') # pass with dataset self.P.read_data(datafile("Ni.dat"), 'X', 25.0, 0.0) self.P.selectAll(self.iset, 'i') self.P.selectAll(self.iset, 'j') # fail for phase 2 self.assertRaises(pdffit2.unassignedError, self.P.selectAll, 2, 'i') self.assertRaises(pdffit2.unassignedError, self.P.selectAll, 2, 'j')