示例#1
0
    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)
示例#3
0
 def test_resource_to_shard_invalid_lane(self):
     # Test rejection of invalid lane number
     with self.assertRaises(AssertionError):
         ShardMask.resource_to_shard('abc', 3)