def test_get_assignment(self): assignment = Assignment() assignment.set_assignment(self.var1, 1) self.assertEqual(1, assignment.get_assignment(self.var1)) self.assertEqual(None, assignment.get_assignment(self.var2))