POCS/pocs/tests/test_dithered_observation.py at 1f1eb629188d5e54f81742a5b9211c688dab492d · panoptes/POCS · 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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
import pytest

from astropy import units as u
from pocs.scheduler.field import Field
from pocs.scheduler.observation.dithered import DitheredObservation as Observation
from pocs.utils import dither


@pytest.fixture
def field():
    return Field('Test Observation', '20h00m43.7135s +22d42m39.0645s')


def test_create_observation_no_field():
    with pytest.raises(TypeError):
        Observation()


def test_create_observation_bad_field():
    with pytest.raises(AssertionError):
        Observation('20h00m43.7135s +22d42m39.0645s')


def test_create_observation_exp_time_no_units(field):
    with pytest.raises(TypeError):
        Observation(field, exp_time=1.0)


def test_create_observation_exp_time_bad(field):
    with pytest.raises(AssertionError):
        Observation(field, exp_time=0.0 * u.second)


def test_create_observation_exp_time_minutes(field):
    obs = Observation(field, exp_time=5.0 * u.minute)
    assert obs.exp_time == 300 * u.second


def test_bad_priority(field):
    with pytest.raises(AssertionError):
        Observation(field, priority=-1)


def test_good_priority(field):
    obs = Observation(field, priority=5.0)
    assert obs.priority == 5.0


def test_priority_str(field):
    obs = Observation(field, priority="5")
    assert obs.priority == 5.0


def test_bad_min_set_combo(field):
    with pytest.raises(AssertionError):
        Observation(field, exp_set_size=7)
    with pytest.raises(AssertionError):
        Observation(field, min_nexp=57)


def test_small_sets(field):
    obs = Observation(field, exp_time=1 * u.second, min_nexp=1, exp_set_size=1)
    assert obs.minimum_duration == 1 * u.second
    assert obs.set_duration == 1 * u.second


def test_good_min_set_combo(field):
    obs = Observation(field, min_nexp=21, exp_set_size=3)
    assert isinstance(obs, Observation)


def test_default_min_duration(field):
    obs = Observation(field)
    assert obs.minimum_duration == 7200 * u.second


def test_default_set_duration(field):
    obs = Observation(field)
    assert obs.set_duration == 1200 * u.second


def test_print(field):
    obs = Observation(field, exp_time=17.5 * u.second, min_nexp=27, exp_set_size=9)
    assert str(obs) == "Test Observation: 17.5 s exposures in blocks of 9, minimum 27, priority 100"


def test_seq_time(field):
    obs = Observation(field, exp_time=17.5 * u.second, min_nexp=27, exp_set_size=9)
    assert obs.seq_time is None


def test_no_exposures(field):
    obs = Observation(field, exp_time=17.5 * u.second, min_nexp=27, exp_set_size=9)
    assert obs.first_exposure is None
    assert obs.last_exposure is None
    assert obs.pointing_image is None


def test_last_exposure_and_reset(field):
    obs = Observation(field, exp_time=17.5 * u.second, min_nexp=27, exp_set_size=9)
    status = obs.status()
    assert status['current_exp'] == obs.current_exp_num

    # Mimic taking exposures
    obs.merit = 112.5

    for i in range(5):
        obs.exposure_list['image_{}'.format(i)] = 'full_image_path_{}'.format(i)

    last = obs.last_exposure
    assert isinstance(last, tuple)
    assert obs.merit > 0.0
    assert obs.current_exp_num == 5

    assert last[0] == 'image_4'
    assert last[1] == 'full_image_path_4'

    assert isinstance(obs.first_exposure, tuple)
    assert obs.first_exposure[0] == 'image_0'
    assert obs.first_exposure[1] == 'full_image_path_0'

    obs.reset()
    status2 = obs.status()

    assert status2['current_exp'] == 0
    assert status2['merit'] == 0.0
    assert obs.first_exposure is None
    assert obs.last_exposure is None
    assert obs.seq_time is None