diff --git a/needle/driver.py b/needle/driver.py index cd10b49..426f38d 100644 --- a/needle/driver.py +++ b/needle/driver.py @@ -61,12 +61,13 @@ def get_screenshot(self): Returns a screenshot of this element as a PIL image. """ d = self.get_dimensions() + devicePixelRatio = int(self._parent.execute_script("return window.devicePixelRatio")) # Cast values to int in order for _ImageCrop not to break d['left'] = int(d['left']) d['top'] = int(d['top']) - d['width'] = int(d['width']) - d['height'] = int(d['height']) + d['width'] = int(d['width']) * devicePixelRatio + d['height'] = int(d['height']) * devicePixelRatio try: # For selenium >= 2.46.1, W3C WebDriver spec drivers (like geckodriver)