def test_y_division_even(self): comm = emp.comm_grid(self.pat1, self.cols1, '', '1,3') assert_array_equal(comm[0][1]['pair'], ['(0.15 0.1) - (0.15 0.2)', '(0.15 0.1) - (0.15 0.3)', '(0.15 0.2) - (0.15 0.3)']) assert_array_almost_equal(comm[0][1]['x'], [0.1, 0.2, 0.1]) assert_array_equal(comm[0][1]['y'], [2/3., 2/3., 1.])
def test_x_y_division_uneven_y(self): comm = emp.comm_grid(self.pat1, self.cols1, '', '2,2') print comm assert_array_equal(comm[0][1]['pair'], ['(0.1 0.125) - (0.1 0.275)', '(0.1 0.125) - (0.2 0.125)', '(0.1 0.125) - (0.2 0.275)', '(0.1 0.275) - (0.2 0.125)', '(0.1 0.275) - (0.2 0.275)', '(0.2 0.125) - (0.2 0.275)']) assert_array_almost_equal(comm[0][1]['x'], [0.15, 0.1, 0.180278, 0.180278, 0.1, 0.15], 6) assert_array_equal(comm[0][1]['y'], [2/3., 0, 0, 0, 2/3., 0])
def test_x_y_division_uneven_y_jaccard(self): comm = emp.comm_grid(self.pat1, self.cols1, '', '2,2',metric='Jaccard') assert_array_equal(comm[0][1]['y'], [1/2., 0, 0, 0, 1/2., 0])
def test_with_split(self): comm = emp.comm_grid(self.pat1, self.cols1, 'year:split', '2,1') assert_array_equal(comm[0][1]['y'], [0]) assert_array_equal(comm[1][1]['y'], [2/3.])
def test_no_splits_Jaccard(self): comm = emp.comm_grid(self.pat1, self.cols1, None, '2,1', metric='Jaccard') assert_almost_equal(comm[0][1]['x'], [0.1]) assert_array_equal(comm[0][1]['y'], [1/2.])
def test_no_splits_Sorensen(self): comm = emp.comm_grid(self.pat1, self.cols1, None, '2,1') assert_almost_equal(comm[0][1]['x'], [0.1]) assert_array_equal(comm[0][1]['y'], [2./(2+1)])