def test_multipe_dataset(): """ generated with: for run in RunInfo.objects.all().filter(problem_categories=10): print("assert '{}' == get_dataset([{}, '{}'])".format( run.type.dataset, run.run_number, run.type.reco) ) """ session = DQMSession() assert "/Cosmics/Run2018E-PromptReco-v1/DQMIO" == session.get_dataset( 325680, "Prompt") assert "/StreamExpress/Run2018D-Express-v1/DQMIO" == session.get_dataset( 324878, "Express") assert "/StreamExpressCosmics/Run2018D-Express-v1/DQMIO" == session.get_dataset( 324231, "Express") assert "/ZeroBias/Run2018D-PromptReco-v2/DQMIO" == session.get_dataset( 323954, "Prompt") assert "/Cosmics/Run2018D-PromptReco-v2/DQMIO" == session.get_dataset( 323109, "Prompt") assert "/ZeroBias/Run2018C-PromptReco-v3/DQMIO" == session.get_dataset( 320065, "Prompt") assert "/StreamExpress/Run2018C-Express-v1/DQMIO" == session.get_dataset( 319909, "Express") assert "/StreamExpressCosmics/Run2018C-Express-v1/DQMIO" == session.get_dataset( 319882, "Express")
def test_get_available_datasets(): dqm_session = DQMSession() datasets = dqm_session.get_available_datasets(321012) assert "/Commissioning/Run2018D-PromptReco-v2/DQMIO" in datasets assert "/StreamExpress/Run2018D-Express-v1/DQMIO" in datasets assert "/StreamExpress/Run2018D-Express-v1/DQMIO" in datasets assert "/ZeroBias/Run2018D-PromptReco-v2/DQMIO" in datasets
def test_heavy_ion(): session = DQMSession() assert "/StreamHIExpress/HIRun2018A-Express-v1/DQMIO" == session.get_dataset( 327564, "Express") assert "/StreamHIExpress/HIRun2018A-Express-v1/DQMIO" == session.get_dataset( 327489, "Express") assert "/HIMinimumBias1/HIRun2018A-PromptReco-v2/DQMIO" == session.get_dataset( 327489, "Prompt")
def test_prompt_with_existing_rereco(): session = DQMSession() assert "/ZeroBias/Run2018A-PromptReco-v1/DQMIO" == session.get_dataset( 315642, "Prompt") assert "/ZeroBias/Run2018B-PromptReco-v1/DQMIO" == session.get_dataset( 317080, "Prompt") assert "/ZeroBias/Run2018B-PromptReco-v1/DQMIO" == session.get_dataset( 317696, "Prompt")
def test_rereco(): session = DQMSession() assert "/SingleTrack/Run2017G-17Nov2017-v1/DQMIO" == session.get_dataset( 306645, "reReco") assert "/SingleTrack/Run2017G-17Nov2017-v1/DQMIO" == session.get_dataset( 306631, "reReco") assert "/SingleTrack/Run2017G-17Nov2017-v1/DQMIO" == session.get_dataset( 306584, "reReco") assert "/ZeroBias/Run2018A-17Sep2018-v1/DQMIO" == session.get_dataset( 315642, "reReco") assert "/ZeroBias/Run2018B-17Sep2018-v1/DQMIO" == session.get_dataset( 317080, "reReco") assert "/ZeroBias/Run2018B-17Sep2018-v1/DQMIO" == session.get_dataset( 317696, "reReco")
class DQMCrawler: def __init__(self, dataset_cache=None): self.dqm_session = DQMSession() if dataset_cache: self.dqm_session.cache.datasets.update(dataset_cache) def get_json(self, run_number, reconstruction, resource): """ :param resource: Name of the resource, e.g. "/Tracking/TrackParameters/generalTracks/GeneralProperties/TrackEtaPhi_ImpactPoint_GenTk" :param run_number: Run number, e.g. 321233 :param reconstruction: Reconstruction type, e.g. "Prompt", "Express" or even "Online" :return: """ reconstruction = reconstruction.lower() if reconstruction == "online": dataset = "/Global/Online/ALL" service = "online" else: dataset = self.dqm_session.get_dataset(run_number, reconstruction) service = "offline" return jsonfairy.get_json(run_number, dataset, resource, service) def get_tracking_map(self, run_number, reconstruction): """ Tries to retrieve the resource TrackEtaPhi_ImpactPoint_GenTk. Works for collisions runs only. :param run_number: Run number :param reconstruction: "Express", "Prompt", "reReco" or "Online" :return: """ resource = _get_tracking_map_resource_name(reconstruction) return self.get_json(run_number, reconstruction, resource)
def test_prompt(): session = DQMSession() assert "/ZeroBias/Run2018A-PromptReco-v3/DQMIO" == session.get_dataset( 316944, "Prompt")
def test_get_lumisections(): assert "534" == DQMSession().get_lumisections(322222, "Prompt")
def test_single_dataset(): assert "/StreamExpress/Run2018D-Express-v1/DQMIO" == DQMSession( ).get_dataset(321012, "Express")
def __init__(self, dataset_cache=None): self.dqm_session = DQMSession() if dataset_cache: self.dqm_session.cache.datasets.update(dataset_cache)