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)])
