PythonProgramming/Week03/test_pyramid.py at 5ae1705487ad7798b1037515a7c00a01499f9ce1 · canbula/PythonProgramming · 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
import os


files = [f for f in os.listdir(os.path.dirname(__file__)) if f.startswith("pyramid")]
for f in files:
    exec("import " + f[:-3] + " as " + f[:-3])
    print(f"The module {f[:-3]} has been imported.")


def test_names():
    for f in files:
        assert "calculate_pyramid_height" in dir(eval(f[:-3])), (
            "calculate_pyramid_height is not defined in " + f[:-3]
        )


def test_types():
    for f in files:
        assert callable(eval(f[:-3]).calculate_pyramid_height), (
            "calculate_pyramid_height is not callable in " + f[:-3]
        )
        assert isinstance(eval(f[:-3]).calculate_pyramid_height(1), int), (
            "calculate_pyramid_height is not returning an int in " + f[:-3]
        )


def test_calculate_pyramid_height():
    for f in files:
        assert eval(f[:-3]).calculate_pyramid_height(1) == 1, (
            "calculate_pyramid_height is not working in " + f[:-3]
        )
        assert eval(f[:-3]).calculate_pyramid_height(2) == 1, (
            "calculate_pyramid_height is not working in " + f[:-3]
        )
        assert eval(f[:-3]).calculate_pyramid_height(6) == 3, (
            "calculate_pyramid_height is not working in " + f[:-3]
        )
        assert eval(f[:-3]).calculate_pyramid_height(20) == 5, (
            "calculate_pyramid_height is not working in " + f[:-3]
        )
        assert eval(f[:-3]).calculate_pyramid_height(100) == 13, (
            "calculate_pyramid_height is not working in " + f[:-3]
        )
        assert eval(f[:-3]).calculate_pyramid_height(1000) == 44, (
            "calculate_pyramid_height is not working in " + f[:-3]
        )
        assert eval(f[:-3]).calculate_pyramid_height(10000) == 140, (
            "calculate_pyramid_height is not working in " + f[:-3]
        )
        assert eval(f[:-3]).calculate_pyramid_height(100000) == 446, (
            "calculate_pyramid_height is not working in " + f[:-3]
        )