def test_bin_no_prefix_probs_bin_out(self):
"""Test binary input without a 0b prefix and binary output."""
in_probs = {
"000": 2 / 7,
"001": 1 / 7,
"010": 1 / 7,
"011": 1 / 7,
"100": 2 / 7
}
probs = ProbDistribution(in_probs)
self.assertEqual(in_probs, probs.binary_probabilities())
def test_hex_probs_bin_out_padded(self):
"""Test hexadecimal input and binary output, padded with zeros."""
in_probs = {
"0x0": 2 / 7,
"0x1": 1 / 7,
"0x2": 1 / 7,
"0x3": 1 / 7,
"0x4": 2 / 7
}
probs = ProbDistribution(in_probs)
expected = {
"0000": 2 / 7,
"0001": 1 / 7,
"0010": 1 / 7,
"0011": 1 / 7,
"0100": 2 / 7
}
self.assertEqual(expected, probs.binary_probabilities(num_bits=4))
def test_bin_probs_bin_out(self):
"""Test binary input and binary output."""
in_probs = {
"0b0": 2 / 7,
"0b1": 1 / 7,
"0b10": 1 / 7,
"0b11": 1 / 7,
"0b100": 2 / 7
}
probs = ProbDistribution(in_probs)
expected = {
"000": 2 / 7,
"001": 1 / 7,
"010": 1 / 7,
"011": 1 / 7,
"100": 2 / 7
}
self.assertEqual(expected, probs.binary_probabilities())
def test_empty_bin_out(self):
"""Test empty input with binary output."""
probs = ProbDistribution({})
self.assertEqual(probs.binary_probabilities(), {})
def test_hex_probs_bin_out(self):
"""Test hexadecimal input and binary output."""
in_probs = {"0x0": 2 / 7, "0x1": 1 / 7, "0x2": 1 / 7, "0x3": 1 / 7, "0x4": 2 / 7}
probs = ProbDistribution(in_probs)
expected = {"0": 2 / 7, "1": 1 / 7, "10": 1 / 7, "11": 1 / 7, "100": 2 / 7}
self.assertEqual(expected, probs.binary_probabilities())
