def test_ratios_for_category(self): """ """ output_dict = ratios_for_category(self.crawford_distance_matrix_headers, self.crawford_distnace_matrix_data, self.crawford_mapping_file_headers, self.crawford_mapping_file_data, 'Treatment') self.assertEquals(output_dict, {'Control': (5, 5), 'Fast': (4,4)}) output_dict = ratios_for_category(self.distance_matrix_headers, self.distance_matrix_data, self.mapping_file_headers, self.mapping_file_data, 'Category') self.assertEquals(output_dict, {'cat_1': (1,3), 'cat_2': (0,2)})
def main(): option_parser, opts, args = parse_command_line_parameters(**script_info) subject_header_name = opts.subject # gradient_header_name = opts.gradient matrix_fp = opts.matrix_fp mapping_fp = opts.mapping_fp matrix_header, matrix_data = parse_distmat(open(matrix_fp, 'U')) mapping_data, mapping_headers, _ = parse_mapping_file(open(mapping_fp, 'U')) out_dict = ratios_for_category(matrix_header, matrix_data, mapping_headers, mapping_data, subject_header_name) print 'Subject\tPercent' for key, value in out_dict.iteritems(): print '%s\t%f' % (key, (value[0]/value[1])*100)