def testAttrCount(self): d1 = Driver('d1') d1.add_attr(key='foo', value='bar1', number=True, subkey='one') d1.add_attr(key='foo', value='bar2', number=True, subkey='two') d1.add_attr(key='foo', value='bar3', number=True, subkey='three') d1.add_attr(key='foo', value='bar4', number=True, subkey='four') self.assertEqual(d1.attr_query(key='foo', number=2, count=True), 1) self.assertEqual(d1.attr_query(key='foo', number=0, count=True), 1) self.assertEqual(d1.attr_query(key='foo', number=False, count=True), 0) self.assertEqual(d1.attr_query(key='foo', count=True), 4) self.assertEqual(d1.attr_query(subkey='four', count=True), 1) d1.del_attrs(key='foo', value='bar1', number=True, subkey='one') d1.add_attr(key='foo', value='bar5', number=True, subkey='five') self.assertEqual(d1.attr_query(key='foo', number=0, count=True), 0) self.assertEqual(d1.attr_query(key='foo', number=4, count=True), 1)