FilesExpand file tree

 submodule/aima-data

/

utils_test.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

108 lines (80 loc) · 2.87 KB

 submodule/aima-data

/

utils_test.py

Copy path

Top

File metadata and controls

• Code
• Blame

108 lines (80 loc) · 2.87 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

import pytest

from utils import *

def test_struct_initialization():

s = Struct(a=1, b=2)

assert s.a == 1

assert s.b == 2

def test_struct_assignment():

s = Struct(a=1)

s.a = 3

assert s.a == 3

def test_update_dict():

assert update({'a': 1}, a=10, b=20) == {'a': 10, 'b': 20}

assert update({}, a=5) == {'a': 5}

def test_update_struct():

assert update(Struct(a=1), a=30, b=20).__cmp__(Struct(a=30, b=20))

assert update(Struct(), a=10).__cmp__(Struct(a=10))

def test_removeall_list():

assert removeall(4, []) == []

assert removeall(4, [1, 2, 3, 4]) == [1, 2, 3]

assert removeall(4, [4, 1, 4, 2, 3, 4, 4] == [1, 2, 3]

def test_removeall_string():

assert removeall('s', '') == ''

assert removeall('s', 'This is a test. Was a test.') == 'Thi i a tet. Wa a tet.'

def test_unique():

assert unique([1, 2, 3, 2, 1]) == [1, 2, 3]

assert unique([1, 5, 6, 7, 6, 5]) == [1, 5, 6, 7]

def test_product():

assert product([1,2,3,4]) == 24

assert product(range(1, 11)) == 3628800

def test_find_if():

assert find_if(callable, [1, 2, 3]) == None

assert find_if(callable, [3, min, max]) == min

def test_count_if():

assert count_if(callable, [42, None, max, min]) == 2

is_odd = lambda x: x % 2

assert count_if(is_odd, []) == 0

assert count_if(is_odd, [1, 2, 3, 4, 5]) == 3

def test_every():

assert every(callable, [min, max]) == 1

assert every(callable, [min, 3]) == 0

def test_some():

assert some(callable, [min, 3]) == 1

assert some(callable, [2, 3]) == 0

def test_isin():

e = []

assert isin(e, [1, e, 3]) == True

assert isin(e, [1, [], 3]) == False

def test_argmin():

assert argmin([-2, 1], lambda x: x**2) == 1

def test_argmin_list():

assert argmin_list(['one', 'to', 'three', 'or'], len) == ['to', 'or']

def test_argmin_gen():

assert [i for i in argmin_gen(['one', 'to', 'three', 'or'], len)] == ['to', 'or']

def test_argmax():

assert argmax([-2, 1], lambda x: x**2) == -2

assert argmax(['one', 'to', 'three'], len) == 'three'

def test_argmax_list():

assert argmax_list(['one', 'three', 'seven'], lambda x: len(x)) == ['three', 'seven']

def test_argmax_gen():

assert argmax_list(['one', 'three', 'seven'], len) == ['three', 'seven']

def test_dotproduct():

assert dotproduct([1, 2, 3], [1000, 100, 10]) == 1230

def test_vector_add():

assert vector_add((0, 1), (8, 9)) == (8, 10)

def test_num_or_str():

assert num_or_str('42') == 42

assert num_or_str(' 42x ') == '42x'

def test_normalize():

assert normalize([1,2,1]) == [0.25, 0.5, 0.25]

def test_clip():

assert [clip(x, 0, 1) for x in [-1, 0.5, 10]] == [0, 0.5, 1]

def test_vector_clip():

assert vector_clip((-1, 10), (0, 0), (9, 9)) == (0, 9)

def test_caller():

assert caller(0) == 'caller'

def f():

return caller()

assert f() == 'f'

if __name__ == '__main__':

pytest.main()