def test_resource_to_shard(self): # Test known addresses addresses = ['abc', 'def', 'XYZ'] shards = [2, 3, 1] for add, sh in zip(addresses, shards): self.assertEqual(ShardMask.resource_to_shard(add, 4), sh) # Repeat for different lane number, expect different results addresses = ['abc', 'def', 'XYZ'] shards = [10, 11, 13] for add, sh in zip(addresses, shards): self.assertEqual(ShardMask.resource_to_shard(add, 16), sh)
def test_resource_to_shard(self): # Test rejection of invalid lane number with self.assertRaises(AssertionError): ShardMask.resource_to_shard('abc', 3) # Test known addresses addresses = ['abc', 'def', 'XYZ'] shards = [2, 3, 1] for add, sh in zip(addresses, shards): self.assertEqual(ShardMask.resource_to_shard(add, 4), sh) # Repeat for different lane number, expect different results addresses = ['abc', 'def', 'XYZ'] shards = [10, 11, 13] for add, sh in zip(addresses, shards): self.assertEqual(ShardMask.resource_to_shard(add, 16), sh)
def test_resource_to_shard_invalid_lane(self): # Test rejection of invalid lane number with self.assertRaises(AssertionError): ShardMask.resource_to_shard('abc', 3)