diff --git a/Tests/test_000_sanity.py b/Tests/test_000_sanity.py index f9c34527c..a30786458 100644 --- a/Tests/test_000_sanity.py +++ b/Tests/test_000_sanity.py @@ -12,7 +12,7 @@ assert PIL.Image.VERSION[:3] == '1.1' # Create an image and do stuff with it. im = PIL.Image.new("1", (100, 100)) assert (im.mode, im.size) == ('1', (100, 100)) -assert len(im.tobytes() if py3 else im.tostring()) == 1300 +assert len(im.tobytes()) == 1300 # Create images in all remaining major modes. im = PIL.Image.new("L", (100, 100)) diff --git a/Tests/test_image_frombytes.py b/Tests/test_image_frombytes.py new file mode 100644 index 000000000..aa157aa6a --- /dev/null +++ b/Tests/test_image_frombytes.py @@ -0,0 +1,10 @@ +from tester import * + +from PIL import Image + +def test_sanity(): + im1 = lena() + im2 = Image.frombytes(im1.mode, im1.size, im1.tobytes()) + + assert_image_equal(im1, im2) + diff --git a/Tests/test_image_fromstring.py b/Tests/test_image_fromstring.py deleted file mode 100644 index ddc53ebcc..000000000 --- a/Tests/test_image_fromstring.py +++ /dev/null @@ -1,16 +0,0 @@ -from tester import * - -from PIL import Image - -def test_sanity(): - - im1 = lena() - im2 = None - - if py3: - im2 = Image.frombytes(im1.mode, im1.size, im1.tobytes()) - else: - im2 = Image.fromstring(im1.mode, im1.size, im1.tostring()) - - assert_image_equal(im1, im2) - diff --git a/Tests/test_image_tostring.py b/Tests/test_image_tobytes.py similarity index 64% rename from Tests/test_image_tostring.py rename to Tests/test_image_tobytes.py index c4d89df93..d42399993 100644 --- a/Tests/test_image_tostring.py +++ b/Tests/test_image_tobytes.py @@ -3,5 +3,5 @@ from tester import * from PIL import Image def test_sanity(): - data = lena().tobytes() if py3 else lena().tostring() + data = lena().tobytes() assert_true(isinstance(data, bytes)) diff --git a/Tests/test_lib_pack.py b/Tests/test_lib_pack.py index c295ac652..5bf622c36 100644 --- a/Tests/test_lib_pack.py +++ b/Tests/test_lib_pack.py @@ -16,7 +16,7 @@ def test_pack(): if py3: return list(im.tobytes("raw", rawmode)) else: - return [ord(c) for c in im.tostring("raw", rawmode)] + return [ord(c) for c in im.tobytes("raw", rawmode)] assert_equal(pack("1", "1"), [128]) assert_equal(pack("1", "1;I"), [0]) @@ -53,10 +53,10 @@ def test_unpack(): if py3: data = bytes(range(1,bytes_+1)) - im = Image.frombytes(mode, (1, 1), data, "raw", rawmode, 0, 1) else: data = ''.join(chr(i) for i in range(1,bytes_+1)) - im = Image.fromstring(mode, (1, 1), data, "raw", rawmode, 0, 1) + + im = Image.frombytes(mode, (1, 1), data, "raw", rawmode, 0, 1) return im.getpixel((0, 0)) @@ -67,7 +67,7 @@ def test_unpack(): if py3: im = Image.frombytes(mode, (8, 1), bytes([value]), "raw", rawmode, 0, 1) else: - im = Image.fromstring(mode, (8, 1), chr(value), "raw", rawmode, 0, 1) + im = Image.frombytes(mode, (8, 1), chr(value), "raw", rawmode, 0, 1) return tuple(im.getdata()) diff --git a/Tests/test_mode_i16.py b/Tests/test_mode_i16.py index 060a68bca..584bdd933 100644 --- a/Tests/test_mode_i16.py +++ b/Tests/test_mode_i16.py @@ -79,18 +79,15 @@ def test_basic(): basic("I") -def test_tostring(): +def test_tobytes(): - def tostring(mode): - if py3: - return Image.new(mode, (1, 1), 1).tobytes() - else: - return Image.new(mode, (1, 1), 1).tostring() + def tobytes(mode): + return Image.new(mode, (1, 1), 1).tobytes() - assert_equal(tostring("L"), b"\x01") - assert_equal(tostring("I;16"), b"\x01\x00") - assert_equal(tostring("I;16B"), b"\x00\x01") - assert_equal(tostring("I"), b"\x01\x00\x00\x00") + assert_equal(tobytes("L"), b"\x01") + assert_equal(tobytes("I;16"), b"\x01\x00") + assert_equal(tobytes("I;16B"), b"\x00\x01") + assert_equal(tobytes("I"), b"\x01\x00\x00\x00") def test_convert(): diff --git a/Tests/tester.py b/Tests/tester.py index 14e4a2d2d..370740cb4 100644 --- a/Tests/tester.py +++ b/Tests/tester.py @@ -144,12 +144,9 @@ def assert_image_equal(a, b, msg=None): failure(msg or "got mode %r, expected %r" % (a.mode, b.mode)) elif a.size != b.size: failure(msg or "got size %r, expected %r" % (a.size, b.size)) - elif py3 and a.tobytes() != b.tobytes(): + elif a.tobytes() != b.tobytes(): failure(msg or "got different content") # generate better diff? - elif not py3 and a.tostring() != b.tostring(): - failure(msg or "got different content") - # same complaint? else: success()