/usr/lib/python2.7/dist-packages/photutils/utils/tests/test_prepare_data.py is in python-photutils 0.2.1-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 | # Licensed under a 3-clause BSD style license - see LICENSE.rst
from __future__ import (absolute_import, division, print_function,
unicode_literals)
import numpy as np
from numpy.testing import assert_allclose
from astropy.tests.helper import pytest
from .. import calculate_total_error
SHAPE = (5, 5)
DATAVAL = 2.
DATA = np.ones(SHAPE) * DATAVAL
MASK = np.zeros_like(DATA, dtype=bool)
MASK[2, 2] = True
ERROR = np.ones(SHAPE)
EFFGAIN = np.ones(SHAPE) * DATAVAL
BACKGROUND = np.ones(SHAPE)
WRONG_SHAPE = np.ones((2, 2))
class TestCalculateTotalError(object):
def test_error_shape(self):
with pytest.raises(ValueError):
calculate_total_error(DATA, error=WRONG_SHAPE,
effective_gain=EFFGAIN)
def test_gain_shape(self):
with pytest.raises(ValueError):
calculate_total_error(DATA, error=ERROR,
effective_gain=WRONG_SHAPE)
@pytest.mark.parametrize('effective_gain', (0, -1))
def test_gain_le_zero(self, effective_gain):
with pytest.raises(ValueError):
calculate_total_error(DATA, error=ERROR,
effective_gain=effective_gain)
def test_gain_scalar(self):
error_tot = calculate_total_error(DATA, error=ERROR,
effective_gain=2.)
assert_allclose(error_tot, np.sqrt(2.) * ERROR)
def test_gain_array(self):
error_tot = calculate_total_error(DATA, error=ERROR,
effective_gain=EFFGAIN)
assert_allclose(error_tot, np.sqrt(2.) * ERROR)
|