def test_get_chain_via_object(self): c = ChainConsumer() c.add_chain(self.data, name="A") c.add_chain(self.data, name="B") assert c._get_chain(c.chains[0])[0] == 0 assert c._get_chain(c.chains[1])[0] == 1 assert len(c._get_chain(c.chains[0])) == 1 assert len(c._get_chain(c.chains[1])) == 1