class getvarExceptions(unittest.TestCase): def setUp(self): self.P = PdfFit() def tearDown(self): del self.P def test_unassignedError(self): """raise pdffit2.unassignedError when variable is undefined""" self.assertRaises(pdffit2.unassignedError, self.P.getvar, self.P.pscale()) def test_ValueError(self): """raise ValueError when a variable index does not exist""" self.P.read_struct(datafile("Ni.stru")) self.assertRaises(ValueError, self.P.getvar, self.P.lat(7))