class TestExpectsAxisLabels(object): def setUp(self): self.obj = ExpectsAxisLabels() self.handler = VerifyWarningHandler() logging.getLogger().addHandler(self.handler) def tearDown(self): root_logger = logging.getLogger() for handler in root_logger.handlers: if isinstance(handler, VerifyWarningHandler): root_logger.handlers.remove(handler) def test_warning(self): self.obj.verify_axis_labels(('a', 'b', 'c'), None, 'foo') assert len(self.handler.records) == 1 assert self.handler.records[0].levelno == logging.WARNING def test_exception(self): assert_raises(ValueError, self.obj.verify_axis_labels, ('a', 'b', 'c'), ('b', 'c', 'd'), 'foo')
def setUp(self): self.obj = ExpectsAxisLabels() self.handler = VerifyWarningHandler() logging.getLogger().addHandler(self.handler)