File size: 5,004 Bytes
2d190aa | 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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 | #!/usr/bin/env python3
"""
Quick test script to verify the demo setup
"""
import sys
def test_imports():
"""Test that all required packages can be imported."""
print("🧪 Testing package imports...")
packages = [
('numpy', 'NumPy'),
('cv2', 'OpenCV'),
('streamlit', 'Streamlit'),
('PIL', 'Pillow'),
('huggingface_hub', 'HuggingFace Hub'),
]
failed = []
for module, name in packages:
try:
__import__(module)
print(f" ✅ {name}")
except ImportError as e:
print(f" ❌ {name}: {e}")
failed.append(name)
return len(failed) == 0, failed
def test_app_structure():
"""Test that app.py has the expected structure."""
print("\n📁 Testing app structure...")
try:
import app
# Check for key functions
functions = [
'load_sample_image',
'downsample_image',
'quantize_image',
'apply_sampling_and_quantization',
'calculate_file_size',
'compress_image_jpeg',
'compress_image_png',
'main_loop',
]
failed = []
for func in functions:
if hasattr(app, func):
print(f" ✅ Function '{func}' found")
else:
print(f" ❌ Function '{func}' missing")
failed.append(func)
return len(failed) == 0, failed
except Exception as e:
print(f" ❌ Error loading app: {e}")
return False, [str(e)]
def test_image_generation():
"""Test that we can generate a sample image."""
print("\n🖼️ Testing image generation...")
try:
import app
import numpy as np
img = app.generate_sample_image(size=256)
# Check image properties
assert img.shape == (256, 256, 3), f"Wrong shape: {img.shape}"
assert img.dtype == np.uint8, f"Wrong dtype: {img.dtype}"
assert img.min() >= 0 and img.max() <= 255, "Invalid pixel values"
print(f" ✅ Generated {img.shape[1]}×{img.shape[0]} image")
return True, []
except Exception as e:
print(f" ❌ Error: {e}")
return False, [str(e)]
def test_image_processing():
"""Test basic image processing functions."""
print("\n⚙️ Testing image processing...")
try:
import app
import numpy as np
# Create a simple test image
test_img = np.random.randint(0, 256, (128, 128, 3), dtype=np.uint8)
# Test downsampling
downsampled = app.downsample_image(test_img, sampling_rate=2)
print(f" ✅ Downsampling works")
# Test quantization
quantized = app.quantize_image(test_img, bits_per_pixel=4)
print(f" ✅ Quantization works")
# Test combined
processed = app.apply_sampling_and_quantization(test_img, 2, 4)
print(f" ✅ Combined processing works")
# Test file size calculation
size = app.calculate_file_size(test_img.shape, 2, 4)
assert size > 0, "File size should be positive"
print(f" ✅ File size calculation works ({size} bytes)")
return True, []
except Exception as e:
print(f" ❌ Error: {e}")
import traceback
traceback.print_exc()
return False, [str(e)]
def main():
"""Run all tests."""
print("=" * 60)
print("🎨 Image Sampling & Quantization Demo - Test Suite")
print("=" * 60)
all_passed = True
# Test 1: Imports
passed, failed = test_imports()
if not passed:
print(f"\n❌ Import test failed. Missing packages: {', '.join(failed)}")
print("\n💡 Run: pip install -r requirements.txt")
all_passed = False
# Test 2: App structure
if passed: # Only run if imports work
passed, failed = test_app_structure()
if not passed:
print(f"\n❌ App structure test failed.")
all_passed = False
# Test 3: Image generation
if passed:
passed, failed = test_image_generation()
if not passed:
print(f"\n❌ Image generation test failed.")
all_passed = False
# Test 4: Image processing
if passed:
passed, failed = test_image_processing()
if not passed:
print(f"\n❌ Image processing test failed.")
all_passed = False
# Summary
print("\n" + "=" * 60)
if all_passed:
print("✅ All tests passed! Ready to run the demo.")
print("\n🚀 Start with: ./run_simple.sh")
print(" Or: make run")
print(" Or: streamlit run app.py")
return 0
else:
print("❌ Some tests failed. Please fix the issues above.")
return 1
if __name__ == "__main__":
sys.exit(main())
|