codingweeks-gameoflife/test_simulation.py at master · lcs27/codingweeks-gameoflife · 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
122
123
124
125
126
127
128
129
130
131
132
133
from simulation import *
from generate_universe import *
import numpy as np


def test_survival():

    # Generation of a universe with seed
    seed = create_seed(type_seed = "r_pentomino")
    universe = generate_universe(size=(6,6))
    universe = add_seed_to_universe(seed, universe,x_start=1, y_start=1)
    '''
    The universe is:
    Po:0,1,2,3,4,5
     0[0,0,0,0,0,0],
     1[0,0,1,1,0,0],
     2[0,1,1,0,0,0],
     3[0,0,1,0,0,0],
     4[0,0,0,0,0,0],
     5[0,0,0,0,0,0]
    '''

    # Reproduction: A dead cell with exactly three live neighbours becomes a live cell
    assert survival((1,1),universe) == 1

    # No change: A dead cell with only 2 live neighbours remains dead
    assert survival((3,3),universe) == 0

    # Generation: A live cell with two or three live neighbours stays alive
    assert survival((1,2),universe) == 1 # 3 live neighbours
    assert survival((1,3),universe) == 1 # 2 live neighbours

    # Overpopulation: A live cell with more than three live neighbours dies
    universe[2,0] = 1
    universe[3,1] = 1
    '''
    The universe is:
    Po:0,1,2,3,4,5
     0[0,0,0,0,0,0],
     1[0,0,1,1,0,0],
     2[1,1,1,0,0,0],
     3[0,1,1,0,0,0],
     4[0,0,0,0,0,0],
     5[0,0,0,0,0,0]
    '''
    assert survival((3,1),universe) == 0

    # Underpopulation: A live cell with fewer than two live neighbours dies
    universe[1,2] = 0
    '''
    The universe is:
    Po:0,1,2,3,4,5
     0[0,0,0,0,0,0],
     1[0,0,0,1,0,0],
     2[1,1,1,0,0,0],
     3[0,1,1,0,0,0],
     4[0,0,0,0,0,0],
     5[0,0,0,0,0,0]
    '''
    assert survival((1,3),universe) == 0


def test_generation():

    # Generation of the next iteration of a universe with pre-added seed
    test_seed_1 = create_seed(type_seed = "r_pentomino")
    test_universe_1 = generate_universe(size=(6, 6))
    test_universe_1 = add_seed_to_universe(test_seed_1, test_universe_1,x_start=1, y_start=1)
    test_universe_1 = generation(test_universe_1)
    '''
    Po:0,1,2,3,4,5
     0[0,0,0,0,0,0],
     1[0,0,1,1,0,0],
     2[0,1,1,0,0,0],
     3[0,0,1,0,0,0],
     4[0,0,0,0,0,0],
     5[0,0,0,0,0,0]
    '''
    test_equality_1 = np.array(test_universe_1 == np.array([[0 , 0, 0, 0, 0, 0],
        [0, 1, 1, 1, 0, 0],
        [0, 1, 0, 0, 0, 0],
        [0, 1, 1, 0, 0, 0],
        [0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0]],dtype=np.uint8))
    assert test_equality_1.all()


def test_simulate():

    # Generation of a universe with seed
    initial_universe = generate_universe(size=(6, 6))
    test_seed_1 = create_seed(type_seed = "r_pentomino")
    initial_universe = add_seed_to_universe(test_seed_1, initial_universe,x_start=1, y_start=1)

    # Zero time
    test_universe_0 = game_life_simulate(initial_universe,0,universe_dict)
    test_equality_0 = np.array(test_universe_0 == np.array([[0 , 0, 0, 0, 0, 0],
        [0 , 0, 1, 1, 0, 0],
        [0 , 1, 1, 0, 0, 0],
        [0 , 0, 1, 0, 0, 0],
        [0 , 0, 0, 0, 0, 0],
        [0 , 0, 0, 0, 0, 0]],dtype=np.uint8))
    assert test_equality_0.all()
    reset_universe_dict(universe_dict)

    # One time
    test_universe_1 = game_life_simulate(initial_universe,1,universe_dict)
    test_equality_1 = np.array(test_universe_1 == np.array([[0 , 0, 0, 0, 0, 0],
        [0 , 1, 1, 1, 0, 0],
        [0 , 1, 0, 0, 0, 0],
        [0 , 1, 1, 0, 0, 0],
        [0 , 0, 0, 0, 0, 0],
        [0 , 0, 0, 0, 0, 0]],dtype=np.uint8))
    assert test_equality_1.all()
    reset_universe_dict(universe_dict)

    # Two times
    test_universe_2 = game_life_simulate(initial_universe,2,universe_dict)
    test_equality_2 = np.array(test_universe_2 == np.array([[0, 0, 1, 0, 0, 0],
        [0, 1, 1, 0, 0, 0],
        [1, 0, 0, 1, 0, 0],
        [0 ,1, 1, 0, 0, 0],
        [0 ,0, 0, 0, 0, 0],
        [0 ,0, 0, 0, 0, 0]],dtype=np.uint8))
    assert test_equality_2.all()
    reset_universe_dict(universe_dict)


### Full Test ###
if __name__ == '__main__':
    test_survival()
    test_generation()
    test_simulate()