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
示例#3
0
 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