Skip to content
Snippets Groups Projects
Unverified Commit e603e35d authored by Piotr Maślanka's avatar Piotr Maślanka Committed by GitHub
Browse files

Merge branch 'develop' into issue-20

parents d696145b aa8d49a9
No related branches found
No related tags found
No related merge requests found
## v2.0.22rc3 ## v2.0.22rc3
* fixes #20 * fixes #20
* fixes #21
## v2.0.22rc2 ## v2.0.22rc2
......
...@@ -232,9 +232,15 @@ class Traceback(object): ...@@ -232,9 +232,15 @@ class Traceback(object):
self.formatted_traceback = six.text_type(traceback.format_exc()) self.formatted_traceback = six.text_type(traceback.format_exc())
def pickle_to(self, stream):
"""Pickle self to target stream"""
pickle.dump(self, stream, pickle.HIGHEST_PROTOCOL)
def pickle(self): def pickle(self):
"""Returns this instance, pickled""" """Returns this instance, pickled"""
return pickle.dumps(self, pickle.HIGHEST_PROTOCOL) bio = io.BytesIO()
self.pickle_to(bio)
return bio.getvalue()
def pretty_format(self): def pretty_format(self):
""" """
......
...@@ -23,7 +23,15 @@ class TestTraceback(unittest.TestCase): ...@@ -23,7 +23,15 @@ class TestTraceback(unittest.TestCase):
p_fmt = tb.pretty_format() p_fmt = tb.pretty_format()
self.assertTrue(p_fmt) self.assertTrue(p_fmt)
print(p_fmt)
def test_issue_21(self):
try:
loc = u'hello world'
raise ValueError(u'hello')
except ValueError:
tb = Traceback()
a = tb.pickle()
self.assertIsInstance(pickle.loads(a), Traceback)
def test_compression_happens(self): def test_compression_happens(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment