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
|
import numpy as np
from drizzle import cdrizzle
def test_cdrizzle():
"""
Call C unit tests for cdrizzle, which are in the src/tests directory
"""
size = 100
data = np.zeros((size, size), dtype='float32')
weights = np.ones((size, size), dtype='float32')
pixmap = np.indices((size, size), dtype='float64')
pixmap = pixmap.transpose()
output_data = np.zeros((size, size), dtype='float32')
output_counts = np.zeros((size, size), dtype='float32')
output_context = np.zeros((size, size), dtype='int32')
cdrizzle.test_cdrizzle(
data,
weights,
pixmap,
output_data,
output_counts,
output_context,
)
|