| |
| """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) |
| |
| |
| print("Testing with small uniform dataset...") |
| data = generate_iid_data(100, 10, 'uniform') |
| results = compress_and_compare(data, "small_test") |
| print_results(results) |
| |
| |
| 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() |
