Coverage for tests/data/make_metricvalue_butlers.py: 15%
62 statements
« prev ^ index » next coverage.py v6.4.1, created at 2022-07-09 06:23 -0700
« prev ^ index » next coverage.py v6.4.1, created at 2022-07-09 06:23 -0700
1#!/usr/bin/env python
2"""Create two butler repos for tests of the metricvalue extract/print code.
3"""
5import astropy.units as u
7import lsst.daf.butler
8from lsst.daf.butler.tests import addDatasetType, addDataIdValue
9from lsst.verify import Measurement
11collection = "testrun"
14def setup_butler(repo):
15 """Create a butler at the given location, and register appropriate things
16 in it to allow adding metricvalues.
17 """
18 lsst.daf.butler.Butler.makeRepo(repo)
19 butler = lsst.daf.butler.Butler(repo, writeable=True)
21 addDataIdValue(butler, "instrument", "TestCam")
22 for x in range(189):
23 addDataIdValue(butler, "detector", x)
24 addDataIdValue(butler, "visit", 12345)
25 addDataIdValue(butler, "visit", 54321)
26 butler.registry.registerCollection(collection, lsst.daf.butler.CollectionType.RUN)
28 return butler
31def add_metricvalues(butler, plus):
32 """Add Measurements as MetricValue datasets to a pre-configured butler,
33 adding ``plus`` to the values that are stored (to allow different repos to
34 have different Measurement values).
35 """
36 dimensions = {"instrument", "visit", "detector"}
37 storageClass = "MetricValue"
38 dataIds = [{"instrument": "TestCam", "visit": 12345, "detector": 12},
39 {"instrument": "TestCam", "visit": 54321, "detector": 25},
40 {"instrument": "TestCam", "visit": 54321, "detector": 12}]
41 addDatasetType(butler, "metricvalue_verify_testing", dimensions, storageClass)
42 value = Measurement("verify.testing", (12 + plus)*u.dimensionless_unscaled)
43 butler.put(value, "metricvalue_verify_testing", dataIds[0], run=collection)
44 value = Measurement("verify.testing", (42 + plus)*u.dimensionless_unscaled)
45 butler.put(value, "metricvalue_verify_testing", dataIds[1], run=collection)
46 value = Measurement("verify.testing", (5 + plus)*u.dimensionless_unscaled)
47 butler.put(value, "metricvalue_verify_testing", dataIds[2], run=collection)
49 addDatasetType(butler, "metricvalue_verify_other", dimensions, storageClass)
50 value = Measurement("verify.other", (7 + plus)*u.ct)
51 butler.put(value, "metricvalue_verify_other", dataIds[0], run=collection)
52 value = Measurement("verify.other", (8 + plus)*u.ct)
53 butler.put(value, "metricvalue_verify_other", dataIds[1], run=collection)
55 addDatasetType(butler, "metricvalue_verify_another", dimensions, storageClass)
56 value = Measurement("verify.another", (3 + plus)*u.mas)
57 butler.put(value, "metricvalue_verify_another", dataIds[0], run=collection)
59 addDatasetType(butler, "metricvalue_verify_testingTime", dimensions, storageClass)
60 value = Measurement("verify.testingTime", (18 + plus)*u.second)
61 butler.put(value, "metricvalue_verify_testingTime", dataIds[0], run=collection)
62 value = Measurement("verify.testingTime", (19 + plus)*u.second)
63 butler.put(value, "metricvalue_verify_testingTime", dataIds[1], run=collection)
65 addDatasetType(butler, "metricvalue_verify_anotherTime", dimensions, storageClass)
66 value = Measurement("verify.anotherTime", (100 + plus)*u.ms)
67 butler.put(value, "metricvalue_verify_anotherTime", dataIds[0], run=collection)
68 value = Measurement("verify.anotherTime", (200 + plus)*u.ms)
69 butler.put(value, "metricvalue_verify_anotherTime", dataIds[1], run=collection)
71 addDatasetType(butler, "metricvalue_verify_testingMemory", dimensions, storageClass)
72 value = Measurement("verify.testingMemory", (100 + plus)*u.Mbyte)
73 butler.put(value, "metricvalue_verify_testingMemory", dataIds[0], run=collection)
74 value = Measurement("verify.testingMemory", (200 + plus)*u.Mbyte)
75 butler.put(value, "metricvalue_verify_testingMemory", dataIds[1], run=collection)
77 addDatasetType(butler, "metricvalue_verify_anotherTaskMemory", dimensions, storageClass)
78 value = Measurement("verify.anotherTaskMemory", (5 + plus)*u.Gbyte)
79 butler.put(value, "metricvalue_verify_anotherTaskMemory", dataIds[0], run=collection)
80 value = Measurement("verify.anotherTaskMemory", (6 + plus)*u.Gbyte)
81 butler.put(value, "metricvalue_verify_anotherTaskMemory", dataIds[1], run=collection)
84if __name__ == "__main__": 84 ↛ 85line 84 didn't jump to line 85, because the condition on line 84 was never true
85 repo = "metricvalue_repo1"
86 butler = setup_butler(repo)
87 add_metricvalues(butler, 0)
89 repo = "metricvalue_repo2"
90 butler = setup_butler(repo)
91 add_metricvalues(butler, 1)