File: test_defaults.py

package info (click to toggle)
python-click 6.6-1~bpo8%2B1
  • links: PTS, VCS
  • area: main
  • in suites: jessie-backports
  • size: 1,068 kB
  • sloc: python: 6,964; makefile: 124
file content (46 lines) | stat: -rw-r--r-- 1,149 bytes parent folder | download | duplicates (11)
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
36
37
38
39
40
41
42
43
44
45
46
import click


def test_basic_defaults(runner):
    @click.command()
    @click.option('--foo', default=42, type=click.FLOAT)
    def cli(foo):
        assert type(foo) is float
        click.echo('FOO:[%s]' % foo)

    result = runner.invoke(cli, [])
    assert not result.exception
    assert 'FOO:[42.0]' in result.output


def test_multiple_defaults(runner):
    @click.command()
    @click.option('--foo', default=[23, 42], type=click.FLOAT,
                  multiple=True)
    def cli(foo):
        for item in foo:
            assert type(item) is float
            click.echo(item)

    result = runner.invoke(cli, [])
    assert not result.exception
    assert result.output.splitlines() == [
        '23.0',
        '42.0',
    ]


def test_nargs_plus_multiple(runner):
    @click.command()
    @click.option('--arg', default=((1, 2), (3, 4)),
                  nargs=2, multiple=True, type=click.INT)
    def cli(arg):
        for item in arg:
            click.echo('<%d|%d>' % item)

    result = runner.invoke(cli, [])
    assert not result.exception
    assert result.output.splitlines() == [
        '<1|2>',
        '<3|4>',
    ]