uvplot/uvplot/tests.py at 3d360748c8302447af4b2040efd105bad63ea425 · mtazzari/uvplot · GitHub

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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#!/usr/bin/env python
# -*- coding: utf-8 -*-

from __future__ import (division, print_function, absolute_import,
                        unicode_literals)

import numpy as np
from numpy.testing import assert_allclose

from uvplot import UVTable
from .uvtable import COLUMNS_V0


def create_sample_uvtable(uvtable_filename):
    u = np.random.randn(10000)*1e4  # m
    v = np.random.randn(10000)*1e4  # m
    re = np.random.randn(10000)  # Jy
    im = np.random.randn(10000)  # Jy
    w = np.random.rand(10000)*1e4

    wle = 0.88e-3  # m

    np.savetxt(uvtable_filename, np.stack((u, v, re, im, w), axis=-1))

    return (u, v, re, im, w), wle


def test_init_uvtable():

    uvtable_filename = "/tmp/uvtable.txt"
    uvtable, wle = create_sample_uvtable(uvtable_filename)
    u, v, re, im, w = uvtable

    # test reading from file
    # format='uvtable'
    uvt_file = UVTable(filename=uvtable_filename, wle=wle, columns=COLUMNS_V0)

    # test providing uvtable tuple
    # takes u, v in units of observing wavelength
    uvt_uvtable1 = UVTable(uvtable=(u, v, re, im, w), wle=wle, columns=COLUMNS_V0)
    uvt_uvtable2 = UVTable(uvtable=(u/wle, v/wle, re, im, w), columns=COLUMNS_V0)

    reference = np.hypot(u/wle, v/wle)

    assert_allclose(uvt_uvtable1.uvdist, reference, atol=0, rtol=1e-16)
    assert_allclose(uvt_uvtable2.uvdist, reference, atol=0, rtol=1e-16)
    assert_allclose(uvt_file.uvdist, reference, atol=0, rtol=1e-16)


def test_deproject():

    uvtable_filename = "/tmp/uvtable.txt"
    uvtable, wle = create_sample_uvtable(uvtable_filename)

    uv = UVTable(filename=uvtable_filename, wle=wle, columns=COLUMNS_V0)

    inc = np.radians(30)
    uv_30 = uv.deproject(inc, inplace=False)

    assert_allclose(uv_30.u, uv.u*np.cos(inc))

    uv.deproject(inc)  # inplace=True by default
    assert_allclose(uv_30.u, uv.u)


def test_uvcut():

    uvtable_filename = "/tmp/uvtable.txt"
    uvtable, wle = create_sample_uvtable(uvtable_filename)

    uv = UVTable(filename=uvtable_filename, wle=wle, columns=COLUMNS_V0)
    uv.header = dict(test='test_header')

    maxuv = 5e3
    uvt = uv.uvcut(maxuv)

    # manual uvcut
    uvcut = uv.uvdist <= maxuv

    assert_allclose(uvt.u, uv.u[uvcut])
    assert_allclose(uvt.v, uv.v[uvcut])
    assert_allclose(uvt.re, uv.re[uvcut])
    assert_allclose(uvt.im, uv.im[uvcut])
    assert_allclose(uvt.weights, uv.weights[uvcut])

    assert uv.header == uvt.header


def test_uvbin():
    uvtable_filename= "/tmp/uvtable.txt"
    uvtable, wle = create_sample_uvtable(uvtable_filename)

    uv = UVTable(filename=uvtable_filename, wle=wle, columns=COLUMNS_V0)

    uvbin_size = 5e4
    uvmin = 3e6
    uvmax = uvmin + uvbin_size

    uv.uvbin(uvbin_size)

    # Check contents of each uvbin agrees with bin_quantity
    bin_re, bin_re_err = uv.bin_quantity(uv.re)

    assert_allclose(bin_re, uv.bin_re)
    assert_allclose(bin_re_err, uv.bin_re_err)

    # Check the contents of one bin is correct
    idx = (uv.uvdist >= uvmin) & (uv.uvdist < uvmax)

    if np.sum(idx > 0):
        assert np.all(uv.bin_index[idx] == uv.bin_index[idx][0])

    id_bin = (uv.bin_uvdist >= uvmin) & (uv.bin_uvdist < uvmax)
    assert np.sum(idx) == np.sum(uv.bin_count[id_bin])