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 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)
