mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2025-04-25 16:10:03 +02:00
017960d963
Some checks failed
Build the package using cmake then documentation / build (ubuntu-latest, 3.12) (push) Failing after 37s
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
import pytest
|
|
import numpy as np
|
|
|
|
from _aare import ClusterVector_Cluster3x3i, Interpolator, Cluster3x3i, ClusterFinder_Cluster3x3i
|
|
|
|
def test_ClusterVector():
|
|
"""Test ClusterVector"""
|
|
|
|
clustervector = ClusterVector_Cluster3x3i()
|
|
assert clustervector.cluster_size_x == 3
|
|
assert clustervector.cluster_size_y == 3
|
|
assert clustervector.item_size() == 4+9*4
|
|
assert clustervector.frame_number == 0
|
|
assert clustervector.capacity == 1024
|
|
assert clustervector.size == 0
|
|
|
|
cluster = Cluster3x3i(0,0,np.ones(9, dtype=np.int32))
|
|
|
|
clustervector.push_back(cluster)
|
|
assert clustervector.size == 1
|
|
|
|
#push_back - check size
|
|
|
|
|
|
|
|
def test_Interpolator():
|
|
"""Test Interpolator"""
|
|
|
|
ebins = np.linspace(0,10, 20, dtype=np.float64)
|
|
xbins = np.linspace(0, 5, 30, dtype=np.float64)
|
|
ybins = np.linspace(0, 5, 30, dtype=np.float64)
|
|
|
|
etacube = np.zeros(shape=[30, 30, 20], dtype=np.float64)
|
|
interpolator = Interpolator(etacube, xbins, ybins, ebins)
|
|
|
|
assert interpolator.get_ietax().shape == (30,30,20)
|
|
assert interpolator.get_ietay().shape == (30,30,20)
|
|
clustervector = ClusterVector_Cluster3x3i()
|
|
|
|
cluster = Cluster3x3i(0,0, np.ones(9, dtype=np.int32))
|
|
#clustervector.push_back(cluster)
|
|
#num_clusters = 1;
|
|
|
|
#assert interpolator.interpolate_Cluster3x3i(clustervector).shape == (num_clusters, 3)
|
|
|
|
|
|
#def test_cluster_file():
|
|
|
|
#def test_cluster_finder():
|
|
#"""Test ClusterFinder"""
|
|
|
|
#clusterfinder = ClusterFinder_Cluster3x3i([100,100])
|
|
|
|
#clusterfinder.find_clusters()
|
|
|
|
#clusters = clusterfinder.steal_clusters()
|
|
|
|
#print("cluster size: ", clusters.size())
|
|
|
|
|
|
|
|
|
|
|
|
|