29 lines
862 B
Python
29 lines
862 B
Python
"""
|
|
Regression tests for functionality that we assume to be working for our analysis.
|
|
"""
|
|
|
|
import pytest
|
|
from sfdata import SFDataFiles, sfdatafile, SFScanInfo
|
|
import tracemalloc
|
|
|
|
|
|
@pytest.mark.regression
|
|
def test_JU_memory():
|
|
base_path = "/sf/cristallina/data/p19739/raw/"
|
|
run_number = 49
|
|
averages = []
|
|
|
|
tracemalloc.start()
|
|
current, peak = tracemalloc.get_traced_memory()
|
|
print(f"Current memory usage is {current / 10**6:.1f}MB; Peak was {peak / 10**6:.1f}MB")
|
|
|
|
|
|
with SFDataFiles(f"{base_path}/run{run_number:04}/data/acq00*.h5") as data:
|
|
ch = data["JF16T03V01"]
|
|
|
|
current, peak = tracemalloc.get_traced_memory()
|
|
print(f"Current memory usage is {current / 10**6:.1f}MB; Peak was {peak / 10**6:.1f}MB")
|
|
tracemalloc.stop()
|
|
|
|
assert current/10**6 < 100, "Memory consumption should be below 100 MB"
|