-
Notifications
You must be signed in to change notification settings - Fork 0
/
derive.py
58 lines (54 loc) · 1.93 KB
/
derive.py
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
def derive(exp):
match exp:
case x if isinstance(x, int):
return 0
case x if isinstance(x, str):
return 1
case (x, '+', y):
return derive(x), '+', derive(y)
case (x, '-', y):
return derive(x), '-', derive(y)
case (x, '*', y):
return (x, '*', derive(y)), '+', (y, '*', derive(x))
case (x, '/', y):
return ((x, '*', derive(y)), '-', (y, '*', derive(x))), '/', (x, '*', x)
case (x, '**', y):
return (y, '*', x, '**', y-1)
case _:
raise ValueError("Derive cannot handle this: " + str(exp))
def simplify(exp):
match exp:
case (x, '+', y) if y == 0:
return x
case (x, '+', y) if x == 0:
return y
case (x, '+', y) if isinstance(x, int) and isinstance(y, int):
return x + y
case (x, '+', y) if isinstance(x, str) and isinstance(y, str):
return 2, '*', x
case (x, '+', y):
return simplify((simplify(x), '+', simplify(y)))
case (x, '*', y) if x == 1:
return y
case (x, '*', y) if y == 1:
return x
case (x, '*', y) if x == 0 or y == 0:
return 0
case (x, '*', y) if isinstance(x, int) and isinstance(y, int):
return x * y
case (x, '*', y) if x == y and isinstance(x, str) and isinstance(y, str):
return x, '**', 2
case (x, '-', y) if isinstance(x, int) and isinstance(y, int):
return x - y
case (x, '-', y) if x == y:
return 0
case (x, '-', y) if x == 0:
return '-', y
case (x, '-', y) if y == 0:
return x
case (x, '**', y) if y == 1:
return x
case (x, '**', y) if y == 0:
return 1
case (x, '**', y) if isinstance(x, int) and isinstance(y, int):
return x ** y