def test_group_median_3(self):
"farray.group_median #3"
sectors = ['a', 'b', 'a', 'b', 'a']
x = np.array([[1, 2], [3, 4], [6, 7], [0, 0], [8, -1]])
desired = np.array([[6.0, 2.0], [1.5, 2.0], [6.0, 2.0], [1.5, 2.0],
[6.0, 2.0]])
actual = group_median(x, sectors)
assert_almost_equal(actual, desired)
def test_group_median_2(self):
"farray.group_median #2"
sectors = ['a', 'b', 'a', 'b', 'a', None]
desired = np.array([[ 2.0, 3.0, 1.5, 0.0, 1.0, nan],
[ 2.0, 0.5, 1.5, nan, nan, nan],
[ 2.0, 3.0, 1.5, 0.0, 1.0, nan],
[ 2.0, 0.5, 1.5, nan, nan, nan],
[ 2.0, 3.0, 1.5, 0.0, 1.0, nan],
[ nan, nan, nan, nan, nan, nan]])
actual = group_median(self.x, sectors)
assert_almost_equal(actual, desired)
def test_median_1(self):
"farray.group_median #1"
sectors = ['a', 'b', 'a', 'b', 'a', 'c']
desired = np.array([[ 2.0, 3.0, 1.5, 0.0, 1.0, nan],
[ 2.0, 0.5, 1.5, nan, nan, nan],
[ 2.0, 3.0, 1.5, 0.0, 1.0, nan],
[ 2.0, 0.5, 1.5, nan, nan, nan],
[ 2.0, 3.0, 1.5, 0.0, 1.0, nan],
[ 5.0, 5.0, 4.0, 4.0, nan, nan]])
actual = group_median(self.x, sectors)
assert_almost_equal(actual, desired)
def test_group_median_2(self):
"farray.group_median #2"
sectors = ['a', 'b', 'a', 'b', 'a', None]
desired = np.array([[2.0, 3.0, 1.5, 0.0, 1.0, nan],
[2.0, 0.5, 1.5, nan, nan, nan],
[2.0, 3.0, 1.5, 0.0, 1.0, nan],
[2.0, 0.5, 1.5, nan, nan, nan],
[2.0, 3.0, 1.5, 0.0, 1.0, nan],
[nan, nan, nan, nan, nan, nan]])
actual = group_median(self.x, sectors)
assert_almost_equal(actual, desired)
def test_median_1(self):
"farray.group_median #1"
sectors = ['a', 'b', 'a', 'b', 'a', 'c']
desired = np.array([[2.0, 3.0, 1.5, 0.0, 1.0, nan],
[2.0, 0.5, 1.5, nan, nan, nan],
[2.0, 3.0, 1.5, 0.0, 1.0, nan],
[2.0, 0.5, 1.5, nan, nan, nan],
[2.0, 3.0, 1.5, 0.0, 1.0, nan],
[5.0, 5.0, 4.0, 4.0, nan, nan]])
actual = group_median(self.x, sectors)
assert_almost_equal(actual, desired)
def test_group_median_3(self):
"farray.group_median #3"
sectors = ['a', 'b', 'a', 'b', 'a']
x = np.array([[1,2],
[3,4],
[6,7],
[0,0],
[8,-1]])
desired = np.array([[6.0,2.0],
[1.5,2.0],
[6.0,2.0],
[1.5,2.0],
[6.0,2.0]])
actual = group_median(x, sectors)
assert_almost_equal(actual, desired)
