Coverage for python/lsst/afw/fits/_fitsContinued.py: 83%

21 statements  

« prev     ^ index     » next       coverage.py v6.5.0, created at 2023-02-11 02:44 -0800

1# This file is part of afw. 

2# 

3# Developed for the LSST Data Management System. 

4# This product includes software developed by the LSST Project 

5# (https://www.lsst.org). 

6# See the COPYRIGHT file at the top-level directory of this distribution 

7# for details of code ownership. 

8# 

9# This program is free software: you can redistribute it and/or modify 

10# it under the terms of the GNU General Public License as published by 

11# the Free Software Foundation, either version 3 of the License, or 

12# (at your option) any later version. 

13# 

14# This program is distributed in the hope that it will be useful, 

15# but WITHOUT ANY WARRANTY; without even the implied warranty of 

16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

17# GNU General Public License for more details. 

18# 

19# You should have received a copy of the GNU General Public License 

20# along with this program. If not, see <https://www.gnu.org/licenses/>. 

21 

22__all__ = ['Fits'] 

23 

24from lsst.utils import continueClass 

25from ._fits import (Fits, ImageWriteOptions, ImageCompressionOptions, ImageScalingOptions, 

26 compressionAlgorithmToString, scalingAlgorithmToString) 

27 

28 

29@continueClass 

30class Fits: # noqa: F811 

31 def __enter__(self): 

32 return self 

33 

34 def __exit__(self, cls, exc, traceback): 

35 self.closeFile() 

36 

37 

38@continueClass 

39class ImageWriteOptions: # noqa: F811 

40 def __repr__(self): 

41 return f"{self.__class__.__name__}(compression={self.compression!r}, scaling={self.scaling!r})" 

42 

43 

44@continueClass 

45class ImageCompressionOptions: # noqa: F811 

46 def __repr__(self): 

47 return (f"{self.__class__.__name__}(algorithm={compressionAlgorithmToString(self.algorithm)!r}, " 

48 f"tiles={self.tiles.tolist()!r}, quantizeLevel={self.quantizeLevel:f})") 

49 

50 

51@continueClass 

52class ImageScalingOptions: # noqa: F811 

53 def __repr__(self): 

54 return (f"{self.__class__.__name__}(algorithm={scalingAlgorithmToString(self.algorithm)!r}, " 

55 f"bitpix={self.bitpix}, maskPlanes={self.maskPlanes}, seed={self.seed} " 

56 f"quantizeLevel={self.quantizeLevel}, quantizePad={self.quantizePad}, " 

57 f"fuzz={self.fuzz}, bscale={self.bscale}, bzero={self.bzero})")