Coverage for tests/test_args.py : 36%

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# This file is part of ap_verify.
3#
4# Developed for the LSST Data Management System.
5# This product includes software developed by the LSST Project
6# (http://www.lsst.org).
7# See the COPYRIGHT file at the top-level directory of this distribution
8# for details of code ownership.
9#
10# This program is free software: you can redistribute it and/or modify
11# it under the terms of the GNU General Public License as published by
12# the Free Software Foundation, either version 3 of the License, or
13# (at your option) any later version.
14#
15# This program is distributed in the hope that it will be useful,
16# but WITHOUT ANY WARRANTY; without even the implied warranty of
17# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18# GNU General Public License for more details.
19#
20# You should have received a copy of the GNU General Public License
21# along with this program. If not, see <http://www.gnu.org/licenses/>.
22#
24import shlex
25import unittest
27import lsst.utils.tests
28import lsst.ap.verify.ap_verify as ap_verify
29import lsst.ap.verify.testUtils
32class CommandLineTestSuite(lsst.ap.verify.testUtils.DataTestCase):
33 # DataTestCase's test dataset is needed for successful parsing of the --dataset argument
35 def _parseString(self, commandLine, parser=None):
36 """Tokenize and parse a command line string.
38 Parameters
39 ----------
40 commandLine : `str`
41 a string containing Unix-style command line arguments, but not the
42 name of the program
43 parser : `argparse.ArgumentParser`, optional
44 the parser to use. Defaults to ``ap_verify``'s primary parser.
46 Returns
47 -------
48 parsed : `argparse.Namespace`
49 The parsed command line.
50 """
51 if not parser:
52 parser = ap_verify._ApVerifyParser()
54 return parser.parse_args(shlex.split(commandLine))
56 def testMissingMain(self):
57 """Verify that a command line consisting missing required arguments is rejected.
58 """
59 args = '--id "visit=54123" --output tests/output/foo'
60 with self.assertRaises(SystemExit):
61 self._parseString(args)
63 def testMissingIngest(self):
64 """Verify that a command line consisting missing required arguments is rejected.
65 """
66 args = '--dataset %s' % CommandLineTestSuite.datasetKey
67 with self.assertRaises(SystemExit):
68 self._parseString(args, ap_verify._IngestOnlyParser())
70 def testMinimumMain(self):
71 """Verify that a command line consisting only of required arguments parses correctly.
72 """
73 args = '--dataset %s --output tests/output/foo' % CommandLineTestSuite.datasetKey
74 parsed = self._parseString(args)
75 self.assertEqual(parsed.dataset.datasetRoot,
76 lsst.utils.getPackageDir(CommandLineTestSuite.testDataset))
77 self.assertEqual(parsed.output, "tests/output/foo")
78 self.assertEqual(parsed.dataIds, [])
80 def testMinimumIngest(self):
81 """Verify that a command line consisting only of required arguments parses correctly.
82 """
83 args = '--dataset %s --output tests/output/foo' % CommandLineTestSuite.datasetKey
84 parsed = self._parseString(args, ap_verify._IngestOnlyParser())
85 self.assertEqual(parsed.dataset.datasetRoot,
86 lsst.utils.getPackageDir(CommandLineTestSuite.testDataset))
87 self.assertEqual(parsed.output, "tests/output/foo")
89 def testDataId(self):
90 """Verify that a command line consisting only of required arguments parses correctly.
91 """
92 args = '--dataset %s --output tests/output/foo --id "visit=54123" --id "filter=x"' \
93 % CommandLineTestSuite.datasetKey
94 parsed = self._parseString(args)
95 self.assertEqual(parsed.dataIds, ["visit=54123", "filter=x"])
97 def testBadDataset(self):
98 """Verify that a command line with an unregistered dataset is rejected.
99 """
100 args = '--dataset FooScope --output tests/output/foo --id "visit=54123"'
101 with self.assertRaises(SystemExit):
102 self._parseString(args)
104 def testBadKeyMain(self):
105 """Verify that a command line with unsupported arguments is rejected.
106 """
107 args = '--dataset %s --output tests/output/foo --id "visit=54123" --clobber' \
108 % CommandLineTestSuite.datasetKey
109 with self.assertRaises(SystemExit):
110 self._parseString(args)
112 def testBadKeyIngest(self):
113 """Verify that a command line with unsupported arguments is rejected.
114 """
115 args = '--dataset %s --output tests/output/foo --id "visit=54123"' \
116 % CommandLineTestSuite.datasetKey
117 with self.assertRaises(SystemExit):
118 self._parseString(args, ap_verify._IngestOnlyParser())
121class MemoryTester(lsst.utils.tests.MemoryTestCase):
122 pass
125def setup_module(module):
126 lsst.utils.tests.init()
129if __name__ == "__main__": 129 ↛ 130line 129 didn't jump to line 130, because the condition on line 129 was never true
130 lsst.utils.tests.init()
131 unittest.main()