test map picklability. add map to top level

fsspec · Apr 21, 2016 · c190a89 · c190a89
1 parent 81883a5
commit c190a89
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/s3fs/__init__.py b/s3fs/__init__.py
@@ -1,3 +1,4 @@
 from .core import S3FileSystem, S3File
+from .mapping import S3Map
 
 __version__ = '0.0.4'
diff --git a/s3fs/tests/test_mapping.py b/s3fs/tests/test_mapping.py
@@ -49,3 +49,13 @@ def test_complex_keys(s3):
     print(list(mw))
 
     assert ('x', 1, 2) in mw
+
+
+def test_pickle(s3):
+    d = S3Map(s3, root)
+    d['x'] = b'1'
+
+    import pickle
+    d2 = pickle.loads(pickle.dumps(d))
+
+    assert d2['x'] == b'1'