Coverage for tests/test_brighterFatter.py : 43%

Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1#
2# LSST Data Management System
3# Copyright 2008-2017 AURA/LSST.
4#
5# This product includes software developed by the
6# LSST Project (http://www.lsst.org/).
7#
8# This program is free software: you can redistribute it and/or modify
9# it under the terms of the GNU General Public License as published by
10# the Free Software Foundation, either version 3 of the License, or
11# (at your option) any later version.
12#
13# This program is distributed in the hope that it will be useful,
14# but WITHOUT ANY WARRANTY; without even the implied warranty of
15# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16# GNU General Public License for more details.
17#
18# You should have received a copy of the LSST License Statement and
19# the GNU General Public License along with this program. If not,
20# see <https://www.lsstcorp.org/LegalNotices/>.
21#
23import unittest
24import pickle
25import os
27import lsst.utils.tests
28import lsst.afw.image as afwImage
29import lsst.ip.isr.isrFunctions as isrFunctions
32class BrighterFatterTestCases(lsst.utils.tests.TestCase):
34 def setUp(self):
35 self.filename = "bf_kernel.pkl"
36 kernel = afwImage.ImageF(17, 17)
37 kernel[9, 9, afwImage.LOCAL] = 1
38 kernelPickleString = kernel.getArray().dumps()
39 # kernel.getArray().dump(self.filename) triggers an "unclosed file" warning with numpy 1.13.1
40 with open(self.filename, 'wb') as f:
41 f.write(kernelPickleString)
43 def tearDown(self):
44 os.unlink(self.filename)
46 def testBrighterFatterInterface(self):
47 """Test brighter fatter correction interface using a delta function kernel on a flat image"""
49 image = afwImage.ImageF(100, 100)
50 image.set(100)
51 ref_image = afwImage.ImageF(image, True)
53 mi = afwImage.makeMaskedImage(image)
54 exp = afwImage.makeExposure(mi)
56 with open(self.filename, 'rb') as f:
57 bfKernel = pickle.load(f)
59 isrFunctions.brighterFatterCorrection(exp, bfKernel, 5, 100, False)
60 self.assertImagesEqual(ref_image, image)
63class MemoryTester(lsst.utils.tests.MemoryTestCase):
64 pass
67def setup_module(module):
68 lsst.utils.tests.init()
71if __name__ == "__main__": 71 ↛ 72line 71 didn't jump to line 72, because the condition on line 71 was never true
72 lsst.utils.tests.init()
73 unittest.main()