Coverage for tests/pickleMapper.py: 42%

22 statements  

« prev     ^ index     » next       coverage.py v6.4.4, created at 2022-08-27 02:38 -0700

1# 

2# LSST Data Management System 

3# Copyright 2008, 2009, 2010 LSST Corporation. 

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 <http://www.lsstcorp.org/LegalNotices/>. 

21# 

22 

23import os 

24import lsst.daf.persistence as dafPersist 

25 

26 

27class PickleMapper(dafPersist.Mapper): 

28 

29 def __init__(self, root=None, outPath='', **kwargs): 

30 self.root = root 

31 self.outPath = outPath 

32 

33 def map_x(self, dataId, write): 

34 path = "foo%(ccd)d.pickle" % dataId 

35 path = os.path.join(self.root, self.outPath, path) 

36 return dafPersist.ButlerLocation( 

37 None, None, "PickleStorage", path, {}, self, 

38 dafPersist.Storage.makeFromURI(os.getcwd())) 

39 

40 def map_dt(self, dataId, write): 

41 path = "dt%(ccd)d.pickle" % dataId 

42 path = os.path.join(self.root, self.outPath, path) 

43 return dafPersist.ButlerLocation( 

44 None, None, "PickleStorage", path, {}, self, 

45 dafPersist.Storage.makeFromURI(os.getcwd())) 

46 

47 def map_p1(self, dataId, write): 

48 path = "p1%(ccd)d.pickle" % dataId 

49 path = os.path.join(self.root, self.outPath, path) 

50 return dafPersist.ButlerLocation( 

51 None, None, "PickleStorage", path, {}, self, 

52 dafPersist.Storage.makeFromURI(os.getcwd())) 

53 

54 def map_p2(self, dataId, write): 

55 path = "p2%(ccd)d.pickle" % dataId 

56 path = os.path.join(self.root, self.outPath, path) 

57 return dafPersist.ButlerLocation( 

58 None, None, "PickleStorage", path, {}, self, 

59 dafPersist.Storage.makeFromURI(os.getcwd()))