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)
