File: test_get_filelike.py

package info (click to toggle)
python-imagesize 1.4.1-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 1,692 kB
  • sloc: python: 599; sh: 6; makefile: 3
file content (35 lines) | stat: -rw-r--r-- 763 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
"""
test imagesize.get for filelike object io.BytesIO(raw bytes)

to test:
pytest -k test_get_filelike
"""

import os
import sys

from io import BytesIO
try:
    # For Python 3.0 and later
    from urllib.request import urlopen
except ImportError:
    # Fall back to Python 2's urllib2
    from urllib2 import urlopen

sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
import imagesize


def test_get_filelike():
    """ test_get_filelike. """

    url = 'https://www.tsln.com/wp-content/uploads/2018/10/bears-tsln-101318-3-1240x826.jpg'
    try:
        response = urlopen(url)
        raw = response.read()
    except Exception as exc:
        raise SystemExit(exc)

    file_like = BytesIO(raw)

    assert imagesize.get(file_like) == (1240, 826)