import numpy as np
from justpfm import justpfm

# Write a test PFM file
data_to_write = np.ones((5, 5), dtype="float32")
justpfm.write_pfm(file_name="test.pfm", data=data_to_write)

# Read the test PFM file
read_data = justpfm.read_pfm(file_name="test.pfm")