def assign(df, tax_dict): """Create str assignment based on tax_ids str and starred boolean. """ ids_stars = df.groupby(by=['condensed_id', 'starred']).groups.keys() df['assignment'] = sequtils.compound_assignment(ids_stars, tax_dict) return df
def test02(self): """ test no data """ taxonomy = self.taxonomy compound = lambda x: sequtils.compound_assignment(x, taxonomy) compound_assignments = map(compound, []) self.assertEquals(compound_assignments, [])
def test01(self): """ test basic behavior """ taxonomy = self.taxonomy thisdatadir = self.thisdatadir this_test = sys._getframe().f_code.co_name compound_assignments_ref = path.join(thisdatadir, this_test, 'compound_names.pkl.bz2') compound_assignments_ref = BZ2File(compound_assignments_ref) compound_assignments_ref = cPickle.load(compound_assignments_ref) compound = lambda x: sequtils.compound_assignment(x, taxonomy) compound_assignments = map(compound, self.assignments) self.assertEquals(compound_assignments, compound_assignments_ref)