Python N2V_DataWrapper.__rand_float_coords2D__の例

コード例 #1

ファイル: test_Noise2VoidDataWrapper.py プロジェクト: sommerc/n2v

def test_random_float_coords():
    boxsize = 13
    np.random.seed(1)
    coords = (np.random.rand() * boxsize, np.random.rand() * boxsize)
    np.random.seed(1)
    assert next(N2V_DataWrapper.__rand_float_coords2D__(boxsize)) == coords

    boxsize = 3
    np.random.seed(1)
    coords = (np.random.rand() * boxsize, np.random.rand() * boxsize, np.random.rand() * boxsize)
    np.random.seed(1)
    assert next(N2V_DataWrapper.__rand_float_coords3D__(boxsize)) == coords

コード例 #2

ファイル: test_Noise2VoidDataWrapper.py プロジェクト: zhang-rx/n2v

def test_coord_gen():
    coord_gen = N2V_DataWrapper.__rand_float_coords2D__(13)
    shape = [128, 128]
    for i in range(100):
        coords = next(coord_gen)
        assert coords[0] < shape[0]
        assert coords[1] < shape[1]

    coord_gen = N2V_DataWrapper.__rand_float_coords3D__(4)
    shape = [55, 55, 55]
    for i in range(100):
        coords = next(coord_gen)
        assert coords[0] < shape[0]
        assert coords[1] < shape[1]
        assert coords[2] < shape[2]