| import base64 | |
| import math | |
| import struct | |
| from tests.support.asserts import assert_png | |
| def png_dimensions(screenshot): | |
| assert_png(screenshot) | |
| image = base64.decodestring(screenshot) | |
| width, height = struct.unpack(">LL", image[16:24]) | |
| return int(width), int(height) |