X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=scripts%2Fsnapdiff.py;h=0ed3937a8f8e669accf577c4b06d6472784e3c57;hb=e2d65227b859e5b096b0f3ccd7e690a662740da6;hp=60d9ae97ca740b26b5820f76567a4d414c6055d3;hpb=6d5954b051edc89b3d0c70e7385aae7439dacd91;p=apitrace

diff --git a/scripts/snapdiff.py b/scripts/snapdiff.py
index 60d9ae9..0ed3937 100755
--- a/scripts/snapdiff.py
+++ b/scripts/snapdiff.py
@@ -48,8 +48,15 @@ class Comparer:
     '''Image comparer.'''
 
     def __init__(self, ref_image, src_image, alpha = False):
-        self.ref_im = Image.open(ref_image)
-        self.src_im = Image.open(src_image)
+        if isinstance(ref_image, basestring):
+            self.ref_im = Image.open(ref_image)
+        else:
+            self.ref_im = ref_image
+
+        if isinstance(src_image, basestring):
+            self.src_im = Image.open(src_image)
+        else:
+            self.src_im = src_image
 
         # Ignore
         if not alpha: