1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
try:
import numpy as np
except ImportError:
np = None
from tests.numpy.util import check_numpy
from tests.testcase import BaseTestCase
class NumpyBaseTestCase(BaseTestCase):
client_kwargs = {'settings': {'use_numpy': True}}
@check_numpy
def setUp(self):
super(NumpyBaseTestCase, self).setUp()
def assertArraysEqual(self, first, second):
return self.assertTrue((first == second).all())
def assertArraysListEqual(self, first, second):
self.assertEqual(len(first), len(second))
for x, y in zip(first, second):
self.assertTrue((x == y).all())
|
