xtensor-python/python-module/tests/test_a.py at module · tdegeus/xtensor-python · GitHub

Repository navigation

import unittest

import xt

import numpy as np

class test_a(unittest.TestCase):

"""

def test_mean(self):

a = np.random.random([103, 102, 101])

n = np.mean(a)

x = xt.mean(a)

self.assertTrue(np.allclose(n, x))

def test_average(self):

a = np.random.random([103, 102, 101])

w = np.random.random([103, 102, 101])

n = np.average(a, weights=w)

x = xt.average(a, w)

self.assertTrue(np.allclose(n, x))

def test_average_axes(self):

a = np.random.random([103, 102, 101])

w = np.random.random([103, 102, 101])

axis = int(np.random.randint(0, high=3))

n = np.average(a, weights=w, axis=(axis,))

x = xt.average(a, w, [axis])

self.assertTrue(np.allclose(n, x))

def test_flip(self):

axis = int(np.random.randint(0, high=3))

a = np.random.random([103, 102, 101])

n = np.flip(a, axis)

x = xt.flip(a, axis)

self.assertTrue(np.allclose(n, x))

def test_cos(self):

a = np.random.random([103, 102, 101])

n = np.cos(a)

x = xt.cos(a)

self.assertTrue(np.allclose(n, x))

if __name__ == "__main__":

unittest.main()