Python random_partitionsの例

コード例 #1

ファイル: test_random_partitions.py プロジェクト: xinmeng2011/python_toolbox

def test():
    '''Test the basic workings of `random_partitions`.'''
    
    def assert_correct_members(partitions):
        '''
        Assert that the `partitions` contain exactly all of `r`'s members.
        '''
        members = sequence_tools.flatten(partitions)
        assert len(members) == len(r)
        assert set(members) == set(r)
        
    r = range(10)
    
    for partition_size in range(1, len(r)):
        partitions = random_tools.random_partitions(r, partition_size)
        for partition in partitions[:-1]:
            assert len(partition) == partition_size
        assert len(partitions[-1]) <= partition_size
        assert_correct_members(partitions)
        
    for n_partitions in range(1, len(r)):
        partitions = random_tools.random_partitions(r,
                                                    n_partitions=n_partitions)
        assert len(partitions) == n_partitions
        assert_correct_members(partitions)
        
        
        

コード例 #2

ファイル: test_random_partitions.py プロジェクト: apddup/python_toolbox

def test():
    '''Test the basic workings of `random_partitions`.'''
    
    def assert_correct_members(partitions):
        '''
        Assert that the `partitions` contain exactly all of `r`'s members.
        '''
        members = sequence_tools.flatten(partitions)
        assert len(members) == len(r)
        assert set(members) == set(r)
        
    r = range(10)
    
    for partition_size in range(1, len(r)):
        partitions = random_tools.random_partitions(r, partition_size)
        for partition in partitions[:-1]:
            assert len(partition) == partition_size
        assert len(partitions[-1]) <= partition_size
        assert_correct_members(partitions)
        
    for n_partitions in range(1, len(r)):
        partitions = random_tools.random_partitions(r,
                                                    n_partitions=n_partitions)
        assert len(partitions) == n_partitions
        assert_correct_members(partitions)