def test_get_key_from_file_invalid_key_name(self):
     """Error dictionary is returned with message."""
     data = tasks.get_key_from_file(self.dummy_fname, 'fake_key')
     for k in ('success', 'message'):
         assert k in data
     assert 'data' not in data
     assert data['success'] == False
     assert data['message'].startswith('Could not find')
 def test_get_key_from_file_invalid_filename(self):
     """Error dictionary is returned with message."""
     # As this test intentionally tries to open a non-existent file, which
     # ROOT would by default print an error about, temporarily change the
     # logging level to suppress such a message
     originalErrorIgnoreLevel = ROOT.gErrorIgnoreLevel
     ROOT.gErrorIgnoreLevel = ROOT.kFatal
     data = tasks.get_key_from_file('fake_file.root', self.dummy_keys[0])
     ROOT.gErrorIgnoreLevel = originalErrorIgnoreLevel
     for k in ('success', 'message'):
         assert k in data
     assert 'data' not in data
     assert data['success'] == False
     assert data['message'].startswith('Could not open')
 def test_get_key_from_file(self):
     """Dictionary describing key in file is returned."""
     data = tasks.get_key_from_file(self.dummy_fname, self.dummy_keys[0])
     for k in ('success', 'data'):
         assert k in data
     data_data = data['data']
     assert data['success'] == True
     assert data_data
     for k in ('filename', 'key_name', 'key_title', 'key_class', 'key_data'):
         assert k in data_data
     assert data_data['filename'].endswith(self.dummy_fname)
     assert data_data['key_name'] == self.dummy_keys[0]
     assert data_data['key_title'] == 'Gaussian'
     assert data_data['key_class'] == 'TH1F'