NP_coding-test/q5_test.py at master · dooyoungk/NP_coding-test · 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
def sequence_gen():
    pass

def test_fib():
    fib = sequence_gen(0,1)
    assert next(fib)== 0
    assert next(fib)== 1
    assert next(fib)== 1
    assert next(fib)== 2
    assert next(fib)== 3
    assert next(fib)== 5
    assert next(fib)== 8
    assert [next(fib) for _ in range(10)] == [13, 21, 34, 55, 89, 144, 233, 377, 610, 987]

def test_trib():
    trib = sequence_gen(0,1,1)
    assert next(trib)==  0
    assert next(trib)==  1
    assert next(trib)==  1
    assert next(trib)==  2
    assert next(trib)==  4
    assert next(trib)==  7
    assert next(trib)==  13
    assert next(trib)==  24
    assert next(trib)==  44
    assert next(trib)==  81
    assert next(trib)==  149
    assert next(trib)==  274
    assert next(trib)==  504
    assert next(trib)==  927
    assert next(trib)==  1705

def test_lucas():
    lucas = sequence_gen(2,1)
    assert [next(lucas) for _ in range(100)] == [2, 1, 3, 4, 7, 11, 18, 29, 47, 76, 123, 199, 322, 521, 843, 1364, 2207, 3571, 5778, 9349, 15127, 24476, 39603, 64079, 103682, 167761, 271443, 439204, 710647, 1149851, 1860498, 3010349, 4870847, 7881196, 12752043, 20633239, 33385282, 54018521, 87403803, 141422324, 228826127, 370248451, 599074578, 969323029, 1568397607, 2537720636, 4106118243, 6643838879, 10749957122, 17393796001, 28143753123, 45537549124, 73681302247, 119218851371, 192900153618, 312119004989, 505019158607, 817138163596, 1322157322203, 2139295485799, 3461452808002, 5600748293801, 9062201101803, 14662949395604, 23725150497407, 38388099893011, 62113250390418, 100501350283429, 162614600673847, 263115950957276, 425730551631123, 688846502588399, 1114577054219522, 1803423556807921, 2918000611027443, 4721424167835364, 7639424778862807, 12360848946698171, 20000273725560978, 32361122672259149, 52361396397820127, 84722519070079276, 137083915467899403, 221806434537978679, 358890350005878082, 580696784543856761, 939587134549734843, 1520283919093591604, 2459871053643326447, 3980154972736918051, 6440026026380244498, 10420180999117162549, 16860207025497407047, 27280388024614569596, 44140595050111976643, 71420983074726546239, 115561578124838522882, 186982561199565069121, 302544139324403592003, 489526700523968661124]
    assert next(lucas)==  792070839848372253127
    assert next(lucas)==  1281597540372340914251
    assert next(lucas)==  2073668380220713167378
    assert next(lucas)==  3355265920593054081629
    assert next(lucas)==  5428934300813767249007
    assert next(lucas)==  8784200221406821330636
    assert next(lucas)==  14213134522220588579643
    assert next(lucas)==  22997334743627409910279
    assert next(lucas)==  37210469265847998489922
    assert next(lucas)==  60207804009475408400201
    assert next(lucas)==  97418273275323406890123
    assert next(lucas)==  157626077284798815290324
    assert next(lucas)==  255044350560122222180447
    assert next(lucas)==  412670427844921037470771
    assert next(lucas)==  667714778405043259651218

def test_pointless():
    pointless = sequence_gen(1)
    assert [next(pointless) for _ in range(1000)] == [1] * 1000