File size: 831 Bytes
24c19d8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | #!/usr/bin/env python3
"""Quick test with smaller datasets to verify functionality."""
import numpy as np
from enumerative_coding import EnumerativeEncoder
from entropy_coding import HuffmanEncoder, theoretical_minimum_size
from test_compression import generate_iid_data, compress_and_compare, print_results
def main():
np.random.seed(42)
# Test with a small dataset
print("Testing with small uniform dataset...")
data = generate_iid_data(100, 10, 'uniform')
results = compress_and_compare(data, "small_test")
print_results(results)
# Test with non-uniform distribution
print("\nTesting with small Zipf dataset...")
data = generate_iid_data(100, 10, 'zipf')
results = compress_and_compare(data, "small_zipf_test")
print_results(results)
if __name__ == "__main__":
main() |