def test_count_each_category(self): test_set_dir = resource_filename("patent_parsing_tools.utils.tests", "test_set") [_, _, categories] = fmatrix.get_test_set_with_categories(test_set_dir) cats = {} print len(categories) for i in categories: for j in categories[i]: if j in cats: cats[j] += 1 else: cats[j] = 1 for i in cats.keys(): print "category['" + i + "'] = " + str(cats[i])
def test_get_test_set_with_categories(self): test_set_dir = resource_filename("patent_parsing_tools.utils.tests", "test_set") [_, matrix, _] = fmatrix.get_test_set_with_categories(test_set_dir) self.assertEqual(matrix.shape[0], 100)