class DiscTest(unittest.TestCase): def setUp(self): self.disc = Disc() program1 = Program('program1') self.disc.programs.update({'program1':program1}) def testGetProgram(self): program = self.disc.getProgram('program1') self.assertIsNotNone(program) def testGetProgramThatNotExists(self): with self.assertRaises(ProgramNotInDiscException): self.disc.getProgram('program3')
def setUp(self): self.disc = Disc() program1 = Program('program1') self.disc.programs.update({'program1':program1})