def test_select_probes_proportional(self):
"""tests the select_probes_proportional() function"""
matrix = dm.DataMatrix(10, 4,
[('R%d' % row) for row in range(1, 11)],
[('C%d' % col) for col in range(1, 5)],
[[1, 11, 21, 31],
[2, 12, 22, 32],
[3, 13, 23, 33],
[4, 14, 24, 34],
[5, 15, 25, 35],
[6, 16, 26, 36],
[7, 17, 27, 37],
[8, 18, 28, 38],
[9, 19, 29, 39],
[10, 20, 30, 40]])
selgenes = human.select_probes(matrix, 10, {1:[0, 1], 2: [2, 3]})
self.assertEquals([0, 1, 2, 3, 4], selgenes)
def test_select_probes_proportional(self):
"""tests the select_probes_proportional() function"""
matrix = dm.DataMatrix(2, 2, ['R1', 'R2'], ['C1', 'C2'], [[1, 3], [2, 4]])
selgenes = human.select_probes(matrix, 10, {1:[0, 1]})
self.assertEquals([0, 1], selgenes)