Python loadPoints Examples

Example #1

def test_affine():
    X = loadPoints(Path("tests/fixtures/ref.txt"))
    Y = loadPoints(Path("tests/fixtures/deg.txt"))
    expected = loadXform(Path("tests/fixtures/affine.pickle"))

    P, xform = last(islice(driftAffine(X, Y, w=0.5), 100))

    np.testing.assert_almost_equal(xform.B, expected.B)
    np.testing.assert_almost_equal(xform.t, expected.t)

Example #2

File: test_align.py Project: TheLaueLab/coherent-point-drift

def test_affine():
    X = loadPoints(Path("tests/fixtures/ref.txt"))
    Y = loadPoints(Path("tests/fixtures/deg.txt"))
    expected = loadXform(Path("tests/fixtures/affine.pickle"))

    P, xform = last(islice(driftAffine(X, Y, w=0.5), 100))

    np.testing.assert_almost_equal(xform.B, expected.B)
    np.testing.assert_almost_equal(xform.t, expected.t)

Example #3

def test_global():
    X = loadPoints(Path("tests/fixtures/ref.txt"))
    Y = loadPoints(Path("tests/fixtures/deg.txt"))
    expected = loadXform(Path("tests/fixtures/xform.pickle"))

    P, xform = globalAlignment(X, Y, w=0.5, mirror=True)

    np.testing.assert_almost_equal(xform.R, expected.R)
    np.testing.assert_almost_equal(xform.t, expected.t)
    np.testing.assert_almost_equal(xform.s, expected.s)

Example #4

File: test_align.py Project: TheLaueLab/coherent-point-drift

def test_global():
    X = loadPoints(Path("tests/fixtures/ref.txt"))
    Y = loadPoints(Path("tests/fixtures/deg.txt"))
    expected = loadXform(Path("tests/fixtures/xform.pickle"))

    P, xform = globalAlignment(X, Y, w=0.5, mirror=True)

    np.testing.assert_almost_equal(xform.R, expected.R)
    np.testing.assert_almost_equal(xform.t, expected.t)
    np.testing.assert_almost_equal(xform.s, expected.s)

Example #5

def test_perfect_rigid():
    X = loadPoints(Path("tests/fixtures/ref.txt"))
    P, xform = last(islice(driftRigid(X, X), 100))

    np.testing.assert_almost_equal(xform.R, np.eye(2))
    np.testing.assert_almost_equal(xform.t, np.zeros(2))
    np.testing.assert_almost_equal(xform.s, 1)

Example #6

File: test_align.py Project: TheLaueLab/coherent-point-drift

def test_perfect_rigid():
    X = loadPoints(Path("tests/fixtures/ref.txt"))
    P, xform = last(islice(driftRigid(X, X), 100))

    np.testing.assert_almost_equal(xform.R, np.eye(2))
    np.testing.assert_almost_equal(xform.t, np.zeros(2))
    np.testing.assert_almost_equal(xform.s, 1)

Example #7

File: test_entry.py Project: TheLaueLab/point-alignment

def test_xform_inverse(tmpdir):
    expected = loadPoints(Path("tests/fixtures/ref.txt"))

    # Generate degraded points (without noise)
    args = split(
        "xform tests/fixtures/ref.txt tests/fixtures/xform.pickle --format txt"
    )
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode
    deg = tmpdir.join("deg.txt")
    deg.write_binary(r.stdout)

    # Generate alignment
    args = split("align tests/fixtures/ref.txt '{}' "
                 "--format pickle --scope global".format(str(deg)))
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode
    xform = tmpdir.join("xform.pickle")
    xform.write_binary(r.stdout)

    # Check alignment xform
    args = split("xform '{}' '{}' --format pickle".format(
        str(deg), str(xform)))
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode
    result = loads(r.stdout)
    assert_almost_equal(expected, result)

Example #8

File: test_entry.py Project: TheLaueLab/coherent-point-drift

def test_print(tmpdir):
    expected = loadPoints(Path("tests/fixtures/ref.txt"))

    # Generate degraded points (without noise)
    args = split("xform tests/fixtures/ref.txt tests/fixtures/xform.pickle --format txt")
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode
    deg = tmpdir.join("deg.txt")
    deg.write_binary(r.stdout)

    # Generate alignment
    args = split("align tests/fixtures/ref.txt '{}' --format print".format(str(deg)))
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode

Example #9

File: test_entry.py Project: TheLaueLab/point-alignment

def test_print(tmpdir):
    expected = loadPoints(Path("tests/fixtures/ref.txt"))

    # Generate degraded points (without noise)
    args = split(
        "xform tests/fixtures/ref.txt tests/fixtures/xform.pickle --format txt"
    )
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode
    deg = tmpdir.join("deg.txt")
    deg.write_binary(r.stdout)

    # Generate alignment
    args = split("align tests/fixtures/ref.txt '{}' --format print".format(
        str(deg)))
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode

Example #10

File: test_entry.py Project: TheLaueLab/coherent-point-drift

def test_xform_inverse(tmpdir):
    expected = loadPoints(Path("tests/fixtures/ref.txt"))

    # Generate degraded points (without noise)
    args = split("xform tests/fixtures/ref.txt tests/fixtures/xform.pickle --format txt")
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode
    deg = tmpdir.join("deg.txt")
    deg.write_binary(r.stdout)

    # Generate alignment
    args = split("align tests/fixtures/ref.txt '{}' "
                 "--format pickle --scope global".format(str(deg)))
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode
    xform = tmpdir.join("xform.pickle")
    xform.write_binary(r.stdout)

    # Check alignment xform
    args = split("xform '{}' '{}' --format pickle".format(str(deg), str(xform)))
    r = run(cmd + args, stdout=PIPE, universal_newlines=False)
    assert not r.returncode
    result = loads(r.stdout)
    assert_almost_equal(expected, result)