Replaced instance methods with static methods + lower case method names.

master
Per A Brodtkorb 8 years ago
parent 675392e6ad
commit e86768d48e

@ -12,20 +12,17 @@ import wafo.kdetools.gridding as wkg
class TestKdeTools(unittest.TestCase): class TestKdeTools(unittest.TestCase):
@staticmethod
def setUp(self): def test_gridcount_1d():
# N = 20 # N = 20
# data = np.random.rayleigh(1, size=(N,)) # data = np.random.rayleigh(1, size=(N,))
self.data = array([0.75355792, 0.72779194, 0.94149169, 0.07841119, data = array([0.75355792, 0.72779194, 0.94149169, 0.07841119,
2.32291887, 1.10419995, 0.77055114, 0.60288273, 2.32291887, 1.10419995, 0.77055114, 0.60288273,
1.36883635, 1.74754326, 1.09547561, 1.01671133, 1.36883635, 1.74754326, 1.09547561, 1.01671133,
0.73211143, 0.61891719, 0.75903487, 1.8919469, 0.73211143, 0.61891719, 0.75903487, 1.8919469,
0.72433808, 1.92973094, 0.44749838, 1.36508452]) 0.72433808, 1.92973094, 0.44749838, 1.36508452])
self.x = np.linspace(0, max(self.data) + 1, 10) x = np.linspace(0, max(data) + 1, 10)
def test_gridcount_1D(self):
data, x = self.data, self.x
dx = x[1] - x[0] dx = x[1] - x[0]
c = wkg.gridcount(data, x) c = wkg.gridcount(data, x)
assert_allclose(c, [0.78762626, 1.77520717, 7.99190087, 4.04054449, assert_allclose(c, [0.78762626, 1.77520717, 7.99190087, 4.04054449,
@ -34,7 +31,8 @@ class TestKdeTools(unittest.TestCase):
t = np.trapz(c / dx / len(data), x) t = np.trapz(c / dx / len(data), x)
assert_allclose(t, 0.9803093435140049) assert_allclose(t, 0.9803093435140049)
def test_gridcount_2D(self): @staticmethod
def test_gridcount_2d():
N = 20 N = 20
# data = np.random.rayleigh(1, size=(2, N)) # data = np.random.rayleigh(1, size=(2, N))
data = array([ data = array([
@ -61,7 +59,8 @@ class TestKdeTools(unittest.TestCase):
t = np.trapz(np.trapz(c / (dx**2 * N), x), x) t = np.trapz(np.trapz(c / (dx**2 * N), x), x)
assert_allclose(t, 0.9011618785736376) assert_allclose(t, 0.9011618785736376)
def test_gridcount_3D(self): @staticmethod
def test_gridcount_3d():
N = 20 N = 20
# data = np.random.rayleigh(1, size=(3, N)) # data = np.random.rayleigh(1, size=(3, N))
data = np.array([ data = np.array([
@ -96,7 +95,8 @@ class TestKdeTools(unittest.TestCase):
t = np.trapz(np.trapz(np.trapz(c / dx**3 / N, x), x), x) t = np.trapz(np.trapz(np.trapz(c / dx**3 / N, x), x), x)
assert_allclose(t, 0.5164999727560187) assert_allclose(t, 0.5164999727560187)
def test_gridcount_4D(self): @staticmethod
def test_gridcount_4d():
N = 20 N = 20
# data = np.random.rayleigh(1, size=(2, N)) # data = np.random.rayleigh(1, size=(2, N))

Loading…
Cancel
Save