def test_attachments(self): su = StatusUpdate('foo bar') su.add_attachment('data.dat', 'XXXX') self.assertEqual(len(su.attachments.keys()), 1) self.assertEqual(su.attachments['data.dat'].file.data, 'XXXX') su.remove_attachment('data.dat') self.assertEqual(len(su.attachments.keys()), 0)