def test_marginal_counts(self): counts = {"00": 13, "01": 36, "10": 28, "11": 23} self.assertEqual(tomo.marginal_counts(counts, [0]), {"0": 41, "1": 59}) self.assertEqual(tomo.marginal_counts(counts, [1]), {"0": 49, "1": 51})