def test_dict_based_machine_partition_n_keys_map(self): pmap = DictBasedMachinePartitionNKeysMap() p1 = MulticastEdgePartition(None, "foo") p2 = MulticastEdgePartition(None, "bar") pmap.set_n_keys_for_partition(p1, 1) pmap.set_n_keys_for_partition(p2, 2) assert pmap.n_keys_for_partition(p1) == 1 assert pmap.n_keys_for_partition(p2) == 2
def test_dict_based_machine_partition_n_keys_map(self): pmap = DictBasedMachinePartitionNKeysMap() p1 = OutgoingEdgePartition("foo", None) p2 = OutgoingEdgePartition("bar", None) pmap.set_n_keys_for_partition(p1, 1) pmap.set_n_keys_for_partition(p2, 2) assert pmap.n_keys_for_partition(p1) == 1 assert pmap.n_keys_for_partition(p2) == 2