scanf/scanf_test.py at master · ejeffrey/scanf · GitHub

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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
#!/usr/bin/env python
import scanf
import pytest

def scanf_star(fmt, data, *args, **kw):
    """Wrap a scanf test to also test null conversion"""

    result = scanf.scanf(fmt, data, *args, **kw)
    if result is None:
        return result
    null_fmt = fmt.replace('%', '%*')
    assert scanf.scanf(null_fmt, data, *args, **kw) == ()
    if "%r" not in fmt:
        result_w_rest = scanf.scanf(fmt + r"%r", data, *args, **kw)
        rest_only = scanf.scanf(null_fmt + r"%r", data, *args, **kw)
        assert result_w_rest[-1] == rest_only[0]
    return result


def test_float():
    assert scanf_star("%f", "32.1") == (32.1,)
    assert scanf_star("%f", "+32.1") == (32.1,)
    assert scanf_star("%f", "+032.1") == (32.1,)
    assert scanf_star("%f", "+32.10") == (32.1,)
    assert scanf_star("%e", "32.2abc") == (32.2,)
    assert scanf_star("%e", "-32.2") == (-32.2,)
    assert scanf_star("%g", "32") == (32.,)
    assert scanf_star("%f", ".3") == (0.3,)
    assert scanf_star("%f", "0.3") == (0.3,)

def test_skip_beginning():
    """Note, this behavior is different from C stdlib"""
    assert scanf_star("%e", "abc 321e-1", search=True) == (32.1,)

def test_literals():
    assert scanf_star("The number is:  %d", "The number is: 52") == (52,)
    assert scanf_star("is: %d", "The number   is 52", search=True) is None
    assert scanf_star("is: %d", "The number is: \n 52", search=True) == (52,)
    assert scanf.scanf("%d%% / %d", "80% / 20%") == (80, 20)
    assert scanf.scanf("%d%%s / %d", "80%s / 20%") == (80, 20)
    assert scanf.scanf("^%s", "^caret")[0] == "caret"

def test_char():
    assert scanf_star("%c", "abc") == ('a',)
    assert scanf_star("%3c", "abc") == ('abc',)
    assert scanf_star("%5c", "abc") is None
    assert scanf_star("%12c has %d", "dodecahedron has 12 letters") == ("dodecahedron", 12)
    assert scanf_star("%s", "The first word") == ("The",)
    assert scanf_star("%s", "Including: punctuation") == ("Including:",)

def test_decimal():
    assert scanf_star("%d", "50")[0] == 50
    assert scanf_star("%d", "050")[0] == 50
    assert scanf_star("%d", "0x50")[0] == 0

    assert scanf_star("%u", "-50") is None
    assert scanf_star("%u", "50")[0] == 50
    assert scanf_star("%u", "050")[0] == 50

def test_signed():
    assert scanf_star("%d", "-42")[0] == -42
    assert scanf_star("%d", "+42")[0] == +42
    assert scanf_star("%d %d", "-0 +42") == (0, 42)

def test_hex():
    assert scanf_star("%x", "0x50")[0]  == 0x50
    assert scanf_star("%x", "0X50")[0]  == 0x50
    assert scanf_star("%X", "0x50")[0]  == 0x50
    assert scanf_star("%X", "0X50")[0]  == 0x50
    assert scanf_star("%x", "50")[0]  == 0x50
    assert scanf_star("%X", "50")[0]  == 0x50

def test_octal():
    assert scanf_star("%o", "050")[0] == 0o50
    assert scanf_star("%o", "777")[0] == 0o777
    assert scanf_star("%o", "0o50")[0] == 0o50
    assert scanf_star("%o", "0O50")[0] == 0o50

def test_rest():
    assert scanf.scanf("%d %r", "99 bottles of beer on the wall") == (99, "bottles of beer on the wall")
    assert scanf.scanf("%d %r string", "42 the rest of the string") == None

def test_binary():
    assert scanf_star("%b", "1100")[0] == 12
    assert scanf_star("%b", "0b1100")[0] == 12

def test_integer():
    assert scanf_star("%i", "50")[0] == 50
    assert scanf_star("%i", "0x50")[0] == 0x50
    assert scanf_star("%i", "0o50")[0] == 0o50
    assert scanf_star("%i", "0b1100")[0] == 12

def test_multiple():
    assert scanf_star("%s %s", "hello, world") == ("hello,", "world")

    # Note that this illustrates the different behavior of
    # regular expressions: it is able to match the literal comma
    # rather than the preceding %s consuming it.
    assert scanf_star("%s, %s", "hello, world") == ("hello", "world")

    assert scanf_star("%d - %d", "52 - 11") == (52, 11)
    mac =  scanf_star("%X:%X:%X:%X:%X:%X", "04:23:AB:03:ef:01")
    assert mac == (0x4, 0x23, 0xab, 0x3, 0xef, 0x1)

def test_list():
    assert scanf_star("%[d]", "4 2 2")[0] == [4, 2, 2]
    assert scanf_star("%[i]", "42 0x42 0xfE")[0] == [42, 0x42, 0xfe]
    assert scanf_star("%[s]", "A bunch of words")[0] == "A bunch of words".split()
    assert scanf_star("%[f]", "42 21.2 93.12")[0] == [42,21.2,93.12]
    assert scanf_star("%[d]", "42")[0] == [42]

def test_separators():
    assert scanf_star("%[d,]", "4, 2, 2")[0] == [4, 2, 2]
    assert scanf_star("%[d,]", "4,2,2")[0] == [4, 2, 2]
    assert scanf_star("%[x,]", "0x32,0x12, 0x1. This is the end")[0] == [0x32, 0x12, 0x1]
    assert scanf_star("%[i,]", "0x32,0o10, 0B110, 53, 21")[0] == [0x32, 0o10, 0B110, 53, 21]
    assert scanf_star("%[d.]", "192.168.1.1")[0] == [192, 168, 1, 1]
    assert scanf_star("%[d.]", "192.168.  1.  1")[0] == [192, 168, 1, 1]
    assert scanf_star("%[x:]", "04:42:ab:C0:DF:4A")[0] == [0x04, 0x42, 0xAB, 0xC0, 0xDF, 0x4A]
    assert scanf_star("%[s,]", "comma, separated, values")[0] == ["comma", "separated", "values"]
    assert scanf_star("%[f,]", "42,21.2,93.12")[0] == [42,21.2,93.12]