Skip to content

Commit 2f04398

Browse files
committed
Add standalone sample input for simple_imager, integrate into unit-test.
This makes it possible to test imaging routines without first generating the test-data with CASA.
1 parent b29ddaf commit 2f04398

File tree

5 files changed

+38
-0
lines changed

5 files changed

+38
-0
lines changed

src/fastimgproto/resources/__init__.py

Whitespace-only changes.
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
import os
2+
3+
testdata_dir = __path__[0]
4+
5+
simple_vis_npz_filepath = os.path.join(testdata_dir,
6+
'simple_vis.npz')

tests/test_scripts/__init__.py

Whitespace-only changes.
+32
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
import json
2+
import numpy as np
3+
4+
from click.testing import CliRunner
5+
from fastimgproto.resources.testdata import simple_vis_npz_filepath
6+
from fastimgproto.scripts.simple_imager import ConfigKeys, \
7+
cli as simple_imager_cli
8+
9+
10+
def test_simple_imager():
11+
runner = CliRunner()
12+
with runner.isolated_filesystem():
13+
config_filename = 'imager_conf.json'
14+
output_filename = 'image.npz'
15+
conf = {
16+
ConfigKeys.image_size_pix: 1024,
17+
ConfigKeys.cell_size_arcsec: 3.
18+
}
19+
with open(config_filename, 'w') as f:
20+
json.dump(conf, f)
21+
22+
result = runner.invoke(simple_imager_cli,
23+
[
24+
config_filename,
25+
simple_vis_npz_filepath,
26+
output_filename
27+
])
28+
assert result.exit_code == 0
29+
with open(output_filename, 'rb') as f:
30+
output_data = np.load(f)
31+
assert 'image' in output_data
32+
assert 'beam' in output_data

0 commit comments

Comments
 (0)