self.assertEqual([c.shape[1:] for c in components],
                     [t.shape for t in get_next])