class CurationToolTestCase(TestCase):
def setUp(self):
self.tool = CurationToolAnnotation()
def tearDown(self):
pass
def test_0000_base(self):
self.assertEqual(self.tool.getFlag('test'), None)
def test_0010_add_and_get(self):
flag = MasterCurationFlag()
flag.id = 'test'
self.tool.addFlag(flag)
self.assertEqual(self.tool.getFlag('test'), flag)
def test_0020_keys(self):
ids = ['test1', 'test2', 'test4']
for i in ids:
flag = MasterCurationFlag()
flag.id = i
self.tool.addFlag(flag)
self.assertEqual(self.tool.keys(), ids)
def test_0030_del(self):
ids = ['test1', 'test2', 'test4']
for i in ids:
flag = MasterCurationFlag()
flag.id = i
self.tool.addFlag(flag)
self.tool.delFlag('test2')
self.assertEqual(self.tool.keys(), ['test1', 'test4'])
