def test_inline_databank(self): from data_bank import InlineDataBank InlineDataBank.add_item('email', '*****@*****.**') InlineDataBank.add_item('email', '*****@*****.**') # duplicated adding InlineDataBank.remove_item('email', '*****@*****.**') # invalid removal InlineDataBank.remove_item('email', '*****@*****.**') # valid removal self.assertEqual(len(InlineDataBank.get_data('email')), 2) InlineDataBank.add_item('phone-number', '0912345678') # add data into a new type InlineDataBank.add_item('phone-number', '0987654321') self.assertEqual(len(InlineDataBank.get_types()), 4) self.assertEqual(len(InlineDataBank.get_data('phone-number')), 2) self.assertIsNone(InlineDataBank.get_data('text'))