dkist-processing-test 1.22.0rc2__py3-none-any.whl → 1.23.0rc2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of dkist-processing-test might be problematic. Click here for more details.
- dkist_processing_test/tasks/__init__.py +0 -1
- dkist_processing_test/tasks/fake_science.py +6 -2
- dkist_processing_test/tests/test_tasks.py +9 -22
- dkist_processing_test/workflows/common_tasks.py +2 -2
- dkist_processing_test/workflows/trial_end_to_end.py +3 -3
- {dkist_processing_test-1.22.0rc2.dist-info → dkist_processing_test-1.23.0rc2.dist-info}/METADATA +2 -2
- {dkist_processing_test-1.22.0rc2.dist-info → dkist_processing_test-1.23.0rc2.dist-info}/RECORD +9 -10
- {dkist_processing_test-1.22.0rc2.dist-info → dkist_processing_test-1.23.0rc2.dist-info}/WHEEL +1 -1
- dkist_processing_test/tasks/trial_output_data.py +0 -46
- {dkist_processing_test-1.22.0rc2.dist-info → dkist_processing_test-1.23.0rc2.dist-info}/top_level.txt +0 -0
|
@@ -6,5 +6,4 @@ from dkist_processing_test.tasks.movie import *
|
|
|
6
6
|
from dkist_processing_test.tasks.noop import *
|
|
7
7
|
from dkist_processing_test.tasks.parse import *
|
|
8
8
|
from dkist_processing_test.tasks.quality import *
|
|
9
|
-
from dkist_processing_test.tasks.trial_output_data import *
|
|
10
9
|
from dkist_processing_test.tasks.write_l1 import *
|
|
@@ -10,6 +10,7 @@ from dkist_processing_common.codecs.json import json_encoder
|
|
|
10
10
|
from dkist_processing_common.models.tags import Tag
|
|
11
11
|
from dkist_processing_common.tasks import WorkflowTaskBase
|
|
12
12
|
from dkist_processing_common.tasks.mixin.input_dataset import InputDatasetMixin
|
|
13
|
+
from dkist_service_configuration.logging import logger
|
|
13
14
|
|
|
14
15
|
from dkist_processing_test.models.parameters import TestParameters
|
|
15
16
|
|
|
@@ -63,6 +64,9 @@ class GenerateCalibratedData(WorkflowTaskBase, InputDatasetMixin):
|
|
|
63
64
|
self.write(data=b"123", tags=[Tag.intermediate(), Tag.task("NOT_USED"), Tag.frame()])
|
|
64
65
|
self.write(data=b"123", tags=["FOO"])
|
|
65
66
|
|
|
67
|
+
logger.info(f"Using {self.parameters.value_message = }")
|
|
68
|
+
logger.info(f"Using {self.parameters.file_message = }")
|
|
69
|
+
|
|
66
70
|
with self.apm_task_step("Loop over inputs"):
|
|
67
71
|
count = 1 # keep a running count to increment the dsps repeat number
|
|
68
72
|
for hdu in self.read(tags=Tag.input(), decoder=fits_hdu_decoder):
|
|
@@ -84,8 +88,8 @@ class GenerateCalibratedData(WorkflowTaskBase, InputDatasetMixin):
|
|
|
84
88
|
header["VBICMOSC"] = count
|
|
85
89
|
|
|
86
90
|
# Sneak date-dependent parameter values into header for end-to-end checking
|
|
87
|
-
header["
|
|
88
|
-
header["
|
|
91
|
+
header["CAM_ID"] = self.parameters.file_message
|
|
92
|
+
header["CAMERA"] = self.parameters.value_message
|
|
89
93
|
|
|
90
94
|
output_hdu = fits.PrimaryHDU(data=data, header=header)
|
|
91
95
|
|
|
@@ -23,6 +23,7 @@ from dkist_processing_common.models.constants import BudName
|
|
|
23
23
|
from dkist_processing_common.models.constants import ConstantsBase
|
|
24
24
|
from dkist_processing_common.models.tags import Tag
|
|
25
25
|
from dkist_processing_common.models.task_name import TaskName
|
|
26
|
+
from dkist_processing_common.tasks import TransferTrialData
|
|
26
27
|
from dkist_processing_common.tests.conftest import FakeGQLClient
|
|
27
28
|
from dkist_service_configuration.logging import logger
|
|
28
29
|
|
|
@@ -34,7 +35,6 @@ from dkist_processing_test.tasks.fake_science import GenerateCalibratedData
|
|
|
34
35
|
from dkist_processing_test.tasks.movie import AssembleTestMovie
|
|
35
36
|
from dkist_processing_test.tasks.movie import MakeTestMovieFrames
|
|
36
37
|
from dkist_processing_test.tasks.noop import NoOpTask
|
|
37
|
-
from dkist_processing_test.tasks.trial_output_data import TransferTestTrialData
|
|
38
38
|
from dkist_processing_test.tasks.write_l1 import WriteL1Data
|
|
39
39
|
from dkist_processing_test.tests.conftest import generate_214_l0_fits_frame
|
|
40
40
|
from dkist_processing_test.tests.conftest import S122Headers
|
|
@@ -288,8 +288,8 @@ def test_generate_calibrated_data(
|
|
|
288
288
|
assert "VBICMOSC" in hdu.header
|
|
289
289
|
|
|
290
290
|
# Verify correct date params were used
|
|
291
|
-
assert hdu.header["
|
|
292
|
-
assert hdu.header["
|
|
291
|
+
assert hdu.header["CAM_ID"] == expected_file_message
|
|
292
|
+
assert hdu.header["CAMERA"] == expected_value_message
|
|
293
293
|
|
|
294
294
|
# Verify debug frame was written
|
|
295
295
|
debug_frame_paths = list(task.read(tags=[Tag.debug(), Tag.frame()]))
|
|
@@ -661,7 +661,7 @@ def trial_output_task(recipe_run_id, tmp_path, mocker):
|
|
|
661
661
|
new=FakeGQLClient,
|
|
662
662
|
)
|
|
663
663
|
proposal_id = "test_proposal_id"
|
|
664
|
-
with
|
|
664
|
+
with TransferTrialData(
|
|
665
665
|
recipe_run_id=recipe_run_id,
|
|
666
666
|
workflow_name="workflow_name",
|
|
667
667
|
workflow_version="workflow_version",
|
|
@@ -678,21 +678,14 @@ def trial_output_task(recipe_run_id, tmp_path, mocker):
|
|
|
678
678
|
task.write(debug_file_obj, tags=[Tag.debug(), Tag.frame()])
|
|
679
679
|
file_count += 1
|
|
680
680
|
|
|
681
|
-
# Write an intermediate frame
|
|
682
|
-
|
|
681
|
+
# Write an intermediate frame
|
|
682
|
+
intermediate_file_obj = uuid4().hex.encode("utf8")
|
|
683
683
|
task.write(
|
|
684
|
-
|
|
684
|
+
intermediate_file_obj,
|
|
685
685
|
tags=[Tag.intermediate(), Tag.frame(), Tag.task("DUMMY")],
|
|
686
686
|
)
|
|
687
687
|
file_count += 1
|
|
688
688
|
|
|
689
|
-
# Write an intermediate frame that we don't want to transfer
|
|
690
|
-
intermediate_discard_file_obj = uuid4().hex.encode("utf8")
|
|
691
|
-
task.write(
|
|
692
|
-
intermediate_discard_file_obj,
|
|
693
|
-
tags=[Tag.intermediate(), Tag.frame(), Tag.task("WHO_CARES")],
|
|
694
|
-
)
|
|
695
|
-
|
|
696
689
|
# An output frame
|
|
697
690
|
output_file_obj = uuid4().hex.encode("utf8")
|
|
698
691
|
task.write(output_file_obj, tags=[Tag.output(), Tag.frame()])
|
|
@@ -723,12 +716,6 @@ def trial_output_task(recipe_run_id, tmp_path, mocker):
|
|
|
723
716
|
task.write(quality_report_file_obj, tags=[Tag.output(), Tag.quality_report()])
|
|
724
717
|
file_count += 1
|
|
725
718
|
|
|
726
|
-
# Specifically tagged files
|
|
727
|
-
task.write(uuid4().hex.encode("utf8"), tags=[Tag.frame(), "FOO", "BAR"])
|
|
728
|
-
file_count += 1
|
|
729
|
-
task.write(uuid4().hex.encode("utf8"), tags=[Tag.frame(), "BAZ"])
|
|
730
|
-
file_count += 1
|
|
731
|
-
|
|
732
719
|
# This one won't get transferred
|
|
733
720
|
task.write(uuid4().hex.encode("utf8"), tags=[Tag.frame(), "FOO"])
|
|
734
721
|
|
|
@@ -738,7 +725,7 @@ def trial_output_task(recipe_run_id, tmp_path, mocker):
|
|
|
738
725
|
|
|
739
726
|
def test_transfer_test_trial_data(trial_output_task, mocker):
|
|
740
727
|
"""
|
|
741
|
-
Given: A
|
|
728
|
+
Given: A TransferTrialData task with associated frames
|
|
742
729
|
When: Running the task and building the transfer list
|
|
743
730
|
Then: No errors occur and the transfer list has the correct number of items
|
|
744
731
|
"""
|
|
@@ -748,7 +735,7 @@ def test_transfer_test_trial_data(trial_output_task, mocker):
|
|
|
748
735
|
"dkist_processing_common.tasks.mixin.globus.GlobusMixin.globus_transfer_scratch_to_object_store"
|
|
749
736
|
)
|
|
750
737
|
mocker.patch(
|
|
751
|
-
"
|
|
738
|
+
"dkist_processing_common.tasks.trial_output_data.TransferTrialData.remove_folder_objects"
|
|
752
739
|
)
|
|
753
740
|
|
|
754
741
|
# Just make sure the thing runs with no errors
|
|
@@ -10,6 +10,7 @@ from dkist_processing_common.tasks import SubmitDatasetMetadata
|
|
|
10
10
|
from dkist_processing_common.tasks import Teardown
|
|
11
11
|
from dkist_processing_common.tasks import TransferL0Data
|
|
12
12
|
from dkist_processing_common.tasks import TransferL1Data
|
|
13
|
+
from dkist_processing_common.tasks import TransferTrialData
|
|
13
14
|
from dkist_processing_common.tasks import TrialTeardown
|
|
14
15
|
from dkist_processing_core import Workflow
|
|
15
16
|
|
|
@@ -19,7 +20,6 @@ from dkist_processing_test.tasks import MakeTestMovieFrames
|
|
|
19
20
|
from dkist_processing_test.tasks import ParseL0TestInputData
|
|
20
21
|
from dkist_processing_test.tasks import TestAssembleQualityData
|
|
21
22
|
from dkist_processing_test.tasks import TestQualityL0Metrics
|
|
22
|
-
from dkist_processing_test.tasks import TransferTestTrialData
|
|
23
23
|
from dkist_processing_test.tasks import WriteL1Data
|
|
24
24
|
from dkist_processing_test.tasks.manual import ManualWithoutProvenance
|
|
25
25
|
from dkist_processing_test.tasks.manual import ManualWithProvenance
|
|
@@ -132,7 +132,7 @@ transfer_trial_data = Workflow(
|
|
|
132
132
|
detail="transfer-trial-data",
|
|
133
133
|
workflow_package=__package__,
|
|
134
134
|
)
|
|
135
|
-
transfer_trial_data.add_node(task=
|
|
135
|
+
transfer_trial_data.add_node(task=TransferTrialData, upstreams=None)
|
|
136
136
|
|
|
137
137
|
# SubmitDatasetMetadata Task
|
|
138
138
|
submit_dataset_metadata = Workflow(
|
|
@@ -6,6 +6,7 @@ from dkist_processing_common.tasks import CreateTrialDatasetInventory
|
|
|
6
6
|
from dkist_processing_common.tasks import CreateTrialQualityReport
|
|
7
7
|
from dkist_processing_common.tasks import QualityL1Metrics
|
|
8
8
|
from dkist_processing_common.tasks import TransferL0Data
|
|
9
|
+
from dkist_processing_common.tasks import TransferTrialData
|
|
9
10
|
from dkist_processing_common.tasks import TrialTeardown
|
|
10
11
|
from dkist_processing_core import Workflow
|
|
11
12
|
|
|
@@ -15,7 +16,6 @@ from dkist_processing_test.tasks import MakeTestMovieFrames
|
|
|
15
16
|
from dkist_processing_test.tasks import ParseL0TestInputData
|
|
16
17
|
from dkist_processing_test.tasks import TestAssembleQualityData
|
|
17
18
|
from dkist_processing_test.tasks import TestQualityL0Metrics
|
|
18
|
-
from dkist_processing_test.tasks import TransferTestTrialData
|
|
19
19
|
from dkist_processing_test.tasks import WriteL1Data
|
|
20
20
|
|
|
21
21
|
trial = Workflow(
|
|
@@ -51,7 +51,7 @@ trial.add_node(
|
|
|
51
51
|
|
|
52
52
|
# Output flow
|
|
53
53
|
trial.add_node(
|
|
54
|
-
task=
|
|
54
|
+
task=TransferTrialData,
|
|
55
55
|
upstreams=[
|
|
56
56
|
CreateTrialDatasetInventory,
|
|
57
57
|
CreateTrialAsdf,
|
|
@@ -61,4 +61,4 @@ trial.add_node(
|
|
|
61
61
|
)
|
|
62
62
|
|
|
63
63
|
# goodby
|
|
64
|
-
trial.add_node(task=TrialTeardown, upstreams=
|
|
64
|
+
trial.add_node(task=TrialTeardown, upstreams=TransferTrialData)
|
{dkist_processing_test-1.22.0rc2.dist-info → dkist_processing_test-1.23.0rc2.dist-info}/METADATA
RENAMED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: dkist-processing-test
|
|
3
|
-
Version: 1.
|
|
3
|
+
Version: 1.23.0rc2
|
|
4
4
|
Summary: Example instrument code used by the DKIST science data processing pipelines to test processing infrastructure
|
|
5
5
|
Author-email: NSO / AURA <dkistdc@nso.edu>
|
|
6
6
|
License: BSD-3-Clause
|
|
@@ -12,7 +12,7 @@ Classifier: Programming Language :: Python :: 3
|
|
|
12
12
|
Classifier: Programming Language :: Python :: 3.11
|
|
13
13
|
Requires-Python: >=3.11
|
|
14
14
|
Description-Content-Type: text/x-rst
|
|
15
|
-
Requires-Dist: dkist-processing-common ==10.
|
|
15
|
+
Requires-Dist: dkist-processing-common ==10.4.0rc1
|
|
16
16
|
Requires-Dist: dkist-header-validator ==5.1.1
|
|
17
17
|
Requires-Dist: dkist-service-configuration ==2.2
|
|
18
18
|
Requires-Dist: dkist-fits-specifications ==4.7.0
|
{dkist_processing_test-1.22.0rc2.dist-info → dkist_processing_test-1.23.0rc2.dist-info}/RECORD
RENAMED
|
@@ -3,33 +3,32 @@ dkist_processing_test/config.py,sha256=LF80ReTk0ggJ3eZI4NZRZCgmR6E84adWU5iP6-JN-
|
|
|
3
3
|
dkist_processing_test/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
4
4
|
dkist_processing_test/models/constants.py,sha256=TxHYZHl3YQXN7-Voj7QgjgAajbxxB_qR8tRI9iTdERI,726
|
|
5
5
|
dkist_processing_test/models/parameters.py,sha256=OLrS1O_xZptjdNW882Otx5cHWyd4FH7nzi8fkZnj9GM,1778
|
|
6
|
-
dkist_processing_test/tasks/__init__.py,sha256=
|
|
6
|
+
dkist_processing_test/tasks/__init__.py,sha256=mzbjxWY9fdNKUyfXDC_vXuSaB7G0-iwrumCUlq6keoA,457
|
|
7
7
|
dkist_processing_test/tasks/exercise_numba.py,sha256=XfMVffPUgeiPoxSgo39tIjzFBJKKt3fnIYGaDFyBibc,1437
|
|
8
8
|
dkist_processing_test/tasks/fail.py,sha256=jiOiuqoX_JR5hx81REgvcSvb0GBlVKc9MIXKUO0Nhr4,245
|
|
9
|
-
dkist_processing_test/tasks/fake_science.py,sha256
|
|
9
|
+
dkist_processing_test/tasks/fake_science.py,sha256=-DMjAtQbtTly0PAv78lIgD8T_HVMo5mmBjyLiYQF0Ko,4555
|
|
10
10
|
dkist_processing_test/tasks/high_memory.py,sha256=J3vBsivFkqs8eZZwBPpEORlHnFX1bGsbO3CjllJMRsc,540
|
|
11
11
|
dkist_processing_test/tasks/manual.py,sha256=gjol_EieMbclv0ZDkVqR2Xd7abutali6KNwoqaCJiAI,583
|
|
12
12
|
dkist_processing_test/tasks/movie.py,sha256=uFPWQlRFP5zzzYGT58IXebbG49su2I4AEpN2bIy7vGc,1831
|
|
13
13
|
dkist_processing_test/tasks/noop.py,sha256=txgpa0DYrjYPKBDfwfjHt7X9Dna4yaJeMCabu3p2Zmo,235
|
|
14
14
|
dkist_processing_test/tasks/parse.py,sha256=4bDKnPaWeZkQbUpYWCWk1O-Fjzu6WWRboFef3NFJDMs,2713
|
|
15
15
|
dkist_processing_test/tasks/quality.py,sha256=9QynI9R5mkYueb46wNBUhK62QSd2n791L0_w6nVxBjc,895
|
|
16
|
-
dkist_processing_test/tasks/trial_output_data.py,sha256=T34kGFxyR0IVaYnJhP3s9kv1_ExSNlnxx0n76cN32NU,1553
|
|
17
16
|
dkist_processing_test/tasks/write_l1.py,sha256=lpNyCI3kH_pnL0Wi1LHhNRjRHNxu6q76xccCNHm7uwA,2074
|
|
18
17
|
dkist_processing_test/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
19
18
|
dkist_processing_test/tests/conftest.py,sha256=9H8_tqNKofHnl1IRqYM2-M2_K_TJ3b6oaIfb1dVkbsg,2728
|
|
20
19
|
dkist_processing_test/tests/parameter_models.py,sha256=c-SqCBqHP4KkFs0_Y7HLED4mH4oMCpAYu-rYVlTHy1M,4635
|
|
21
20
|
dkist_processing_test/tests/test_parameters.py,sha256=KGeEUOhtC_l-LN6hjB2jgwUZIWAMBx8wlmEeIja-eKo,4572
|
|
22
|
-
dkist_processing_test/tests/test_tasks.py,sha256=
|
|
21
|
+
dkist_processing_test/tests/test_tasks.py,sha256=woNEolAfahLowR7T9nTaZgNJidkGmGpfU6Xyc4LXbqQ,26857
|
|
23
22
|
dkist_processing_test/tests/test_workflows.py,sha256=NqRkstqcqwH80cKJ1uDw16G4MsDOuOiZRWdfH0STarQ,286
|
|
24
23
|
dkist_processing_test/workflows/__init__.py,sha256=Ryn_HXsVkZM_lLKyVteXC909Td2geylmz_1Z-8SE8c8,78
|
|
25
|
-
dkist_processing_test/workflows/common_tasks.py,sha256=
|
|
24
|
+
dkist_processing_test/workflows/common_tasks.py,sha256=UpUTt-QhD-jkzBB3p1ECRbtqY3wjP6Lfsy4TdvxZKQ8,6837
|
|
26
25
|
dkist_processing_test/workflows/end_to_end.py,sha256=XzSe7XtPF0e3fLXWL90LOZ0WKCoCKA4SL43N0CzgpCQ,2328
|
|
27
26
|
dkist_processing_test/workflows/exercise_numba.py,sha256=phpPYcIXEsmc0X1yls_YbXoKS6VXxhHe_8vNd_GyajU,347
|
|
28
27
|
dkist_processing_test/workflows/fail.py,sha256=KGhyAF7gKYYutP4aGa-1tDqPsC7Nr4xKYY0z-HByWi0,326
|
|
29
28
|
dkist_processing_test/workflows/noop.py,sha256=k2-6BpRkl1JDGaHPavxDzIlVx11KQtxKmsHmHTNQ9o0,666
|
|
30
29
|
dkist_processing_test/workflows/resource_queue.py,sha256=_jOT99O0MFN_ACne6i8T5ZIwe_lBQqGz2Ccb8JlqQMI,500
|
|
31
|
-
dkist_processing_test/workflows/trial_end_to_end.py,sha256=
|
|
32
|
-
dkist_processing_test-1.
|
|
33
|
-
dkist_processing_test-1.
|
|
34
|
-
dkist_processing_test-1.
|
|
35
|
-
dkist_processing_test-1.
|
|
30
|
+
dkist_processing_test/workflows/trial_end_to_end.py,sha256=4QSphBegbetDben5PdA5_J64hWJVV8ciVdF90qkyZCo,2415
|
|
31
|
+
dkist_processing_test-1.23.0rc2.dist-info/METADATA,sha256=K5m8u0183_VdfQvAlolrafX43mumoDgnNzwaSrfHOqY,3298
|
|
32
|
+
dkist_processing_test-1.23.0rc2.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
|
|
33
|
+
dkist_processing_test-1.23.0rc2.dist-info/top_level.txt,sha256=Hs4oTIrG_r-svhk_RGFTEO4e3vqQoYlBzdv5mvJVF24,22
|
|
34
|
+
dkist_processing_test-1.23.0rc2.dist-info/RECORD,,
|
|
@@ -1,46 +0,0 @@
|
|
|
1
|
-
"""Trial tasks"""
|
|
2
|
-
from dkist_processing_common.tasks.mixin.globus import GlobusTransferItem
|
|
3
|
-
from dkist_processing_common.tasks.trial_output_data import TransferTrialDataBase
|
|
4
|
-
from dkist_service_configuration.logging import logger
|
|
5
|
-
|
|
6
|
-
__all__ = ["TransferTestTrialData"]
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
class TransferTestTrialData(TransferTrialDataBase):
|
|
10
|
-
@property
|
|
11
|
-
def intermediate_task_names(self) -> list[str]:
|
|
12
|
-
"""Return a list of intermediate tasks we want to transfer.
|
|
13
|
-
|
|
14
|
-
Just a dummy task for testing.
|
|
15
|
-
"""
|
|
16
|
-
return ["DUMMY"]
|
|
17
|
-
|
|
18
|
-
def build_transfer_list(self) -> list[GlobusTransferItem]:
|
|
19
|
-
"""
|
|
20
|
-
Build a list containing all files we want to transfer to the trial environment.
|
|
21
|
-
|
|
22
|
-
For the purposes of testing we try to exercise all of the provided helper methods.
|
|
23
|
-
"""
|
|
24
|
-
transfer_list = []
|
|
25
|
-
|
|
26
|
-
transfer_list += self.build_debug_frame_transfer_list()
|
|
27
|
-
|
|
28
|
-
transfer_list += self.build_intermediate_frame_transfer_list()
|
|
29
|
-
|
|
30
|
-
transfer_list += self.build_output_frame_transfer_list()
|
|
31
|
-
|
|
32
|
-
transfer_list += self.build_output_dataset_inventory_transfer_list()
|
|
33
|
-
|
|
34
|
-
transfer_list += self.build_output_asdf_transfer_list()
|
|
35
|
-
|
|
36
|
-
transfer_list += self.build_output_quality_data_transfer_list()
|
|
37
|
-
|
|
38
|
-
transfer_list += self.build_output_quality_report_transfer_list()
|
|
39
|
-
|
|
40
|
-
transfer_list += self.build_output_movie_transfer_list()
|
|
41
|
-
|
|
42
|
-
transfer_list += self.build_transfer_list_from_tag_lists([["FOO", "BAR"], ["BAZ"]])
|
|
43
|
-
|
|
44
|
-
logger.info(f"{transfer_list = }")
|
|
45
|
-
|
|
46
|
-
return transfer_list
|
|
File without changes
|