def test_group_mean_3(self): "farray.group_mean #3" sectors = ['a', 'b', 'a', 'b', 'a'] x = np.array([[1, 2], [3, 4], [6, 7], [0, 0], [8, -1]]) desired = np.array([[5.0, 8 / 3.0], [1.5, 2.0], [5.0, 8 / 3.0], [1.5, 2.0], [5.0, 8 / 3.0]]) actual = group_mean(x, sectors) assert_almost_equal(actual, desired)
def test_group_mean_2(self): "farray.group_mean #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_mean(self.x, sectors) assert_almost_equal(actual, desired)
def test_group_mean_2(self): "farray.group_mean #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_mean(self.x, sectors) assert_almost_equal(actual, desired)
def test_group_mean_1(self): "farray.group_mean #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]]) with np.errstate(invalid='ignore'): actual = group_mean(self.x, sectors) assert_almost_equal(actual, desired)
def test_group_mean_3(self): "farray.group_mean #3" sectors = ['a', 'b', 'a', 'b', 'a'] x = np.array([[1,2], [3,4], [6,7], [0,0], [8,-1]]) desired = np.array([[5.0, 8/3.0], [1.5, 2.0], [5.0, 8/3.0], [1.5, 2.0], [5.0, 8/3.0]]) actual = group_mean(x, sectors) assert_almost_equal(actual, desired)