FilesExpand file tree

 master

/

test_pyarray.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

89 lines (72 loc) · 3.14 KB

 master

/

test_pyarray.py

Copy path

Top

File metadata and controls

• Code
• Blame

89 lines (72 loc) · 3.14 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

############################################################################

# Copyright (c) 2016, Johan Mabille and Sylvain Corlay #

# #

# Distributed under the terms of the BSD 3-Clause License. #

# #

# The full license is in the file LICENSE, distributed with this software. #

############################################################################

import os

import sys

import subprocess

# Build the test extension

here = os.path.abspath(os.path.dirname(__file__))

subprocess.check_call([sys.executable, os.path.join(here, 'setup.py'), 'build_ext', '--inplace'], cwd=here)

# Test it!

from unittest import TestCase

import xtensor_python_test as xt

import numpy as np

class ExampleTest(TestCase):

def test_example1(self):

self.assertEqual(4, xt.example1([4, 5, 6]))

def test_example2(self):

x = np.array([[0., 1.], [2., 3.]])

res = np.array([[2., 3.], [4., 5.]])

y = xt.example2(x)

np.testing.assert_allclose(y, res, 1e-12)

def test_vectorize(self):

x1 = np.array([[0, 1], [2, 3]])

x2 = np.array([0, 1])

res = np.array([[0, 2], [2, 4]])

y = xt.vectorize_example1(x1, x2)

np.testing.assert_array_equal(y, res)

def test_readme_example1(self):

v = np.arange(15).reshape(3, 5)

y = xt.readme_example1(v)

np.testing.assert_allclose(y, 1.2853996391883833, 1e-12)

def test_complex_overload_reg(self):

a = 23.23

c = 2.0 + 3.1j

self.assertEqual(xt.complex_overload_reg(a), a)

self.assertEqual(xt.complex_overload_reg(c), c)

def test_complex_overload(self):

a = np.random.rand(3, 3)

b = np.random.rand(3, 3)

c = a + b * 1j

y = xt.complex_overload(c)

np.testing.assert_allclose(np.imag(y), np.imag(c))

np.testing.assert_allclose(np.real(y), np.real(c))

x = xt.complex_overload(b)

self.assertEqual(x.dtype, b.dtype)

np.testing.assert_allclose(x, b)

def test_readme_example2(self):

x = np.arange(15).reshape(3, 5)

y = [1, 2, 3, 4, 5]

z = xt.readme_example2(x, y)

np.testing.assert_allclose(z,

[[-0.540302, 1.257618, 1.89929 , 0.794764, -1.040465],

[-1.499227, 0.136731, 1.646979, 1.643002, 0.128456],

[-1.084323, -0.583843, 0.45342 , 1.073811, 0.706945]], 1e-5)

def test_rect_to_polar(self):

x = np.ones(10, dtype=complex)

z = xt.rect_to_polar(x[::2]);

np.testing.assert_allclose(z, np.ones(5, dtype=float), 1e-5)

def test_shape_comparison(self):

x = np.ones([4, 4])

y = np.ones([5, 5])

z = np.zeros([4, 4])

self.assertFalse(xt.compare_shapes(x, y))

self.assertTrue(xt.compare_shapes(x, z))

def test_int_overload(self):

for dtype in [np.uint8, np.int8, np.uint16, np.int16, np.uint32, np.int32, np.uint64, np.int64]:

b = xt.int_overload(np.ones((10), dtype))

self.assertEqual(str(dtype.__name__), b)