make classes picklable

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/test_pickle.py b/tests/test_pickle.py
new file mode 100644
index 0000000..cdb698d
--- /dev/null
+++ b/tests/test_pickle.py
@@ -0,0 +1,15 @@
+import unittest
+from pyisda.legs import ContingentLeg
+from pickle import dumps, loads
+import datetime
+
+class TestPickle(unittest.TestCase):
+
+    def test_contingentleg(self):
+        cl = ContingentLeg(datetime.date(2016, 11, 3),
+                           datetime.date(2021, 12, 20),
+                           12,
+                           True)
+        pickle = dumps(cl)
+        import pdb; pdb.set_trace()
+        cl2 = loads(pickle)