def capture_log(self, level=logging.DEBUG): th = TestHandler() th.setLevel(level) logger.addHandler(th) if self._test_handler is not None: l = len(self._test_handler.buffer) yield th.buffer if self._test_handler is not None: self._test_handler.buffer = self._test_handler.buffer[:l]
def test_warnings(self): ureg = UnitRegistry() th = TestHandler() logger.addHandler(th) add_ctxs(ureg) d = Context('ab') ureg.add_context(d) self.assertEqual(len(th.buffer), 1) self.assertIn("ab", str(th.buffer[-1]['message'])) d = Context('ab1', aliases=('ab', )) ureg.add_context(d) self.assertEqual(len(th.buffer), 2) self.assertIn("ab", str(th.buffer[-1]['message']))
def test_warnings(self): ureg = UnitRegistry() th = TestHandler() logger.addHandler(th) add_ctxs(ureg) d = Context('ab') ureg.add_context(d) self.assertEqual(len(th.buffer), 1) self.assertIn("ab", str(th.buffer[-1]['message'])) d = Context('ab1', aliases=('ab',)) ureg.add_context(d) self.assertEqual(len(th.buffer), 2) self.assertIn("ab", str(th.buffer[-1]['message']))
def setUp(self): self._test_handler = None if self.CHECK_NO_WARNING: self._test_handler = th = TestHandler() th.setLevel(logging.WARNING) logger.addHandler(th)
else: import codecs string_types = basestring def u(x): return codecs.unicode_escape_decode(x)[0] from pint import logger, UnitRegistry h = logging.StreamHandler() f = logging.Formatter( "%(levelname)s %(asctime)s %(funcName)s %(lineno)d %(message)s") h.setLevel(logging.DEBUG) h.setFormatter(f) logger.addHandler(h) logger.setLevel(logging.DEBUG) class TestCase(unittest.TestCase): @classmethod def setUpClass(cls): cls.ureg = UnitRegistry(force_ndarray=cls.FORCE_NDARRAY) cls.Q_ = cls.ureg.Quantity def assertSequenceEqual(self, seq1, seq2, msg=None, seq_type=None): if isinstance(seq1, self.Q_): if isinstance(seq2, self.Q_): seq2 = seq2.to(seq1).magnitude seq1 = seq1.magnitude else:
string_types = str def u(x): return x else: import codecs string_types = basestring def u(x): return codecs.unicode_escape_decode(x)[0] from pint import logger, UnitRegistry h = logging.StreamHandler() f = logging.Formatter("%(levelname)s %(asctime)s %(funcName)s %(lineno)d %(message)s") h.setLevel(logging.DEBUG) h.setFormatter(f) logger.addHandler(h) logger.setLevel(logging.DEBUG) class TestCase(unittest.TestCase): @classmethod def setUpClass(cls): cls.ureg = UnitRegistry(force_ndarray=cls.FORCE_NDARRAY) cls.Q_ = cls.ureg.Quantity def assertSequenceEqual(self, seq1, seq2, msg=None, seq_type=None): if isinstance(seq1, self.Q_): if isinstance(seq2, self.Q_): seq2 = seq2.to(seq1).magnitude seq1 = seq1.magnitude