def METHOD_NAME(self):
    for inter, dist in zip(
        self.interactions, self.normalised_state_distribution
    ):
        self.assertEqual(
            dist,
            axl.interaction_utils.compute_normalised_state_distribution(
                inter
            ),
        )