import os
from imgshape.shape import get_shape

def test_get_shape():
    img_path = "assets/sample_images/image_created_with_a_mobile_phone.png"
    if not os.path.exists(img_path):
        print("❌ test.jpg not found in assets/sample_images/")
        return

    shape = get_shape(img_path)
    assert isinstance(shape, tuple)
    assert len(shape) == 3
    assert all(isinstance(x, int) for x in shape)
    print(f"✅ Shape Test Passed: {shape}")

# Run test when executed directly
if __name__ == "__main__":
    test_get_shape()