Working identity test

Author: feynmanliang

labelbox/lbx.py

@@ -9,7 +9,7 @@ def encode(image):
     Returns:
         A `io.BytesIO` containing the LBX encoded image.
     """
-    pass
+    return image
 
 def decode(lbx):
     """Decodes a `io.BytesIO` with LBX encoded data into a `PIL.Image`.
@@ -20,4 +20,4 @@ def decode(lbx):
     Returns:
         A `PIL.Image` of the decoded data.
     """
-    pass
+    return lbx

tests/test_lbx.py

@@ -1,8 +1,11 @@
-from PIL import Image
+from io import BytesIO
+
 import labelbox.lbx as lbx
+import numpy as np
+from PIL import Image
 
 def test_identity(datadir):
-    im = Image.open(datadir.join('PNG_transparency_demonstration_2.png'))
-    print(im)
-    assert np.array(lbx.decode(lbx.encode(im))) == np.array(im)
+    with open(datadir.join('PNG_transparency_demonstration_2.png'), 'rb') as f:
+        im = Image.open(BytesIO(f.read()))
+    assert np.all(np.array(lbx.decode(lbx.encode(im))) == np.array(im))