示例#1
0
def test_partial_key_match():
    test_dict = {
        ('a', 'b', 'c'): (1, 2, 3),
        ('a', 'b', 'd'): (4, 5, 6),
        ('e', 'f', 'g'): (8, 7, 9)
    }
    assert list(util.partial_key_match(
        ('a', None, 'c'), test_dict))[0] == test_dict[('a', 'b', 'c')]
示例#2
0
 def _get_source_id(self, key):
     """
     Returns source_id based on the key.
     """
     source_id_list = list(partial_key_match(key, self.data_sources))
     if len(source_id_list) != 1:
         raise KeyError(f"The values used: {key} do not correspond to one source_id "
                        f"but {len(source_id_list)} source_id(s)."
                        " Please check the list using HelioviewerClient.data_sources.")
     return source_id_list[0]
示例#3
0
 def _get_source_id(self, key):
     """
     Returns source_id based on the key.
     """
     source_id_list = list(partial_key_match(key, self.data_sources))
     if len(source_id_list) > 1:  # or maybe != 1
         raise KeyError(f"The values used: {key} do not correspond to one source_id "
                        f"but {len(source_id_list)} source_id(s)."
                        " Please check the list using HelioviewerClient.data_sources.")
     return source_id_list[0]