FilesExpand file tree

 main

/

bindpy_02.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

80 lines (60 loc) · 1.75 KB

 main

/

bindpy_02.py

Copy path

Top

File metadata and controls

• Code
• Blame

80 lines (60 loc) · 1.75 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

from lpython import i32, f64, pythoncall, Const

from numpy import empty, int32, float64

@pythoncall(module = "bindpy_02_module")

def get_cpython_version() -> str:

pass

@pythoncall(module = "bindpy_02_module")

def get_int_array_sum(a: i32[:]) -> i32:

pass

@pythoncall(module = "bindpy_02_module")

def get_int_array_product(a: i32[:]) -> i32:

pass

@pythoncall(module = "bindpy_02_module")

def get_float_array_sum(a: f64[:]) -> f64:

pass

@pythoncall(module = "bindpy_02_module")

def get_float_array_product(a: f64[:]) -> f64:

pass

@pythoncall(module = "bindpy_02_module")

def show_array_dot_product(a: i32[:], b: f64[:]):

pass

# Integers:

def test_array_ints():

n: Const[i32] = 5

a: i32[n] = empty([n], dtype=int32)

i: i32

for i in range(n):

a[i] = i + 10

assert get_int_array_sum(a) == i32(60)

assert get_int_array_product(a) == i32(240240)

# Floats

def test_array_floats():

n: Const[i32] = 3

m: Const[i32] = 5

b: f64[n, m] = empty([n, m], dtype=float64)

i: i32

j: i32

for i in range(n):

for j in range(m):

b[i, j] = f64((i + 1) * (j + 1))

assert abs(get_float_array_sum(b) - (90.000000)) <= 1e-4

assert abs(get_float_array_product(b) - (13436928000.000000)) <= 1e-4

def test_array_broadcast():

n: Const[i32] = 3

m: Const[i32] = 5

a: i32[n] = empty([n], dtype=int32)

b: f64[n, m] = empty([n, m], dtype=float64)

i: i32

j: i32

for i in range(n):

a[i] = i + 10

for i in range(n):

for j in range(m):

b[i, j] = f64((i + 1) * (j + 1))

show_array_dot_product(a, b)

def main0():

print("CPython version: ", get_cpython_version())

test_array_ints()

test_array_floats()

test_array_broadcast()

main0()