def test_ignore(self): ignore = ['sNp', 'a1'] flag_cnames = {'SNP': 'SNP', 'ASDF': 'ASDF', 'N': 'FOOBAR'} x = munge.get_cname_map(flag_cnames, munge.default_cnames, ignore) # check that ignore columns are ignored nose.tools.assert_raises(KeyError, x.__getitem__, 'SNP') nose.tools.assert_raises(KeyError, x.__getitem__, 'A1') # check that flag columns make it into the dict self.assertEqual(x['ASDF'], 'ASDF') # check that default columns make it into the dict self.assertEqual(x['A2'], 'A2') # check that flags override default self.assertEqual(x['N'], 'FOOBAR')
def test_no_flags(self): x = munge.get_cname_map({}, munge.default_cnames, []) self.assertEqual(x, munge.default_cnames)