Function test_old_module_example2

tests/python/test_docexamples_f32.py:474–500

()

def test_old_module_example2():
    # Create an empty Range Encoder:
    encoder = constriction.stream.queue.RangeEncoder()

    # Same made up data and entropy models as in the ANS Coding example above:
    model = constriction.stream.model.QuantizedGaussian(-100, 100)
    symbols = np.array([23, -15, 78, 43, -69], dtype=np.int32)
    means = np.array([35.2, -1.7, 30.1, 71.2, -75.1], dtype=np.float32)
    stds = np.array([10.1, 25.3, 23.8, 35.4, 3.9], dtype=np.float32)

    # Encode the data (this time in normal order, since Range Coding is a queue):
    encoder.encode(symbols, model, means, stds)

    print(f"Compressed size: {encoder.num_bits()} bits")

    # Get the compressed bit string (sealed up to full words):
    compressed = encoder.get_compressed()

    # ... writing and reading from file same as above (skipped here) ...

    # Initialize a Range Decoder from the compressed bit string:
    decoder = constriction.stream.queue.RangeDecoder(compressed)

    # Decode the data and verify it's correct:
    reconstructed = decoder.decode(model, means, stds)
    assert decoder.maybe_exhausted()
    assert np.all(reconstructed == symbols)


def test_ans_example():