def test_output2():
    out = io.BytesIO()
    segno.make_qr('Good Times', error='M').save(out, kind='png', scale=10, color='red')
    f = tempfile.NamedTemporaryFile('w', suffix='.png', delete=False)
    f.close()
    cli.main(['-e=M', '--scale=10', '--color=red', '--output={0}'.format(f.name), 'Good', 'Times'])
    f = open(f.name, 'rb')
    content = f.read()
    f.close()
    os.unlink(f.name)
    assert out.getvalue() == content

def test_background_set():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='pdf', background='yellow')
    graphic = _find_graphic(out)
    assert 'rg' in graphic
    assert 're' in graphic

def test_background_none():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='pdf')
    graphic = _find_graphic(out)
    assert 'rg' not in graphic
    assert 're' not in graphic

def test_scale_float():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    scale = 1.34
    scale_cmd = '{0} 0 0 {0} 0 0 cm'.format(scale)
    qr.save(out, kind='pdf', scale=scale, compresslevel=0)
    assert scale_cmd in _find_graphic(out)

def test_default_scale():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    scale = 1
    scale_cmd = '{0} 0 0 {0} 0 0 cm'.format(scale)
    qr.save(out, kind='pdf', compresslevel=0)
    assert scale_cmd not in _find_graphic(out)

def test_data_svg_minimal_encoding():
    qr = segno.make_qr('A')
    val = qr.svg_data_uri(encode_minimal=True)
    assert val
    expected = "data:image/svg+xml;charset=utf-8,%3Csvg xmlns='"
    assert expected == val[:len(expected)]
    assert val.endswith('%3C/svg%3E')

def test_write_xpm_indicator():
    qr = segno.make_qr('test', error='m', boost_error=False)
    out = io.StringIO()
    qr.save(out, kind='xpm')
    expected = '/* XPM */\n'
    val = out.getvalue()
    assert expected == val[:len(expected)]

def test_no_line_class_empty_str():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='svg', lineclass='')
    root = _parse_xml(out)
    path_el = _get_path(root)
    assert 'class' not in path_el.attrib

def test_custom_svg_class():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='svg', svgclass='test-class')
    root = _parse_xml(out)
    assert 'class' in root.attrib
    assert 'test-class' == root.attrib.get('class')

def create(content, error='H', version=None, mode=None, encoding=None):

    def translate_error(pyqrcode_error):
        try:
            if pyqrcode_error.upper() in ('L', 'M', 'Q', 'H'):
                return pyqrcode_error
        except AttributeError:
            pass
        try:
            return _ERROR_LEVEL[pyqrcode_error]
        except KeyError:
            raise ValueError('Unknown error level "{0}"'.format(pyqrcode_error))

    def translate_mode(pyqrcode_mode):
        mode = pyqrcode_mode
        try:
            mode = pyqrcode_mode.lower()
        except AttributeError:
            pass
        return mode if mode != 'binary' else 'byte'

    return PyQRCode(segno.make_qr(content, error=translate_error(error),
                                  version=version,
                                  mode=translate_mode(mode),
                                  encoding=encoding, boost_error=False),
                    content, encoding)

def test_data_svg_no_charset():
    qr = segno.make_qr('A')
    val = qr.svg_data_uri(omit_charset=True)
    assert val
    expected = "data:image/svg+xml,%3Csvg%20xmlns%3D%27"
    assert expected == val[:len(expected)]
    assert val.endswith('%3C%2Fsvg%3E')

def test_svgid():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    ident = 'svgid'
    qr.save(out, kind='svg', svgid=ident)
    root = _parse_xml(out)
    assert 'id' in root.attrib
    assert ident == root.attrib['id']

def test_write_svg_background_rgba_svg2():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='svg', background='#0000ffcc', svgversion=2.0)
    assert b'fill-opacity' not in out.getvalue()
    root = _parse_xml(out)
    path = _get_path(root)
    assert path.attrib['fill'].startswith('rgba')

def test_dpi():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='png', dpi=300)
    out.seek(0)
    assert b'pHYs' in out.getvalue()
    # pHYs 11811 (11811 meters = 300 dpi / 0.0254)
    assert b'\x70\x48\x59\x73\x00\x00\x2E\x23\x00\x00\x2E\x23\x01\x78\xA5\x3F\x76' in out.getvalue()

def test_write_txt():
    # Test with default options
    qr = segno.make_qr('test', error='m', boost_error=False)
    out = io.StringIO()
    qr.save(out, kind='txt', border=0)
    expected = '11111110011'
    val = out.getvalue()
    assert expected == val[:len(expected)]

def test_svgversion():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    version = 1.0
    qr.save(out, kind='svg', svgversion=version)
    root = _parse_xml(out)
    assert 'version' in root.attrib
    assert str(version) == root.attrib['version']

def test_save_png_buffer():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='png')
    out.seek(0)
    expected = b'\211PNG\r\n\032\n'  # PNG magic number
    val = out.read(len(expected))
    assert expected == val

def test_viewbox():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='svg', omitsize=True)
    root = _parse_xml(out)
    assert 'viewBox' in root.attrib
    assert 'height' not in root.attrib
    assert 'width' not in root.attrib

def test_title3():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='svg', title='点')
    root = _parse_xml(out)
    title_el = _get_title(root)
    assert title_el is not None
    assert '点' == title_el.text

def test_scale():
    qr = segno.make_qr('test')
    out = io.BytesIO()
    qr.save(out, kind='svg', scale=2)
    root = _parse_xml(out)
    path = _get_path(root)
    assert path is not None
    assert 'scale(2)' in path.attrib['transform']