Exemplo n.º 1
0
def test_cartesianvector_dist_sq():
    v = CartesianVector(1, -4)

    d = v.dist_sq()
    assert d == 1 + 4 * 4

    v2 = CartesianVector(0, -2)
    d = v.dist_sq(v2)
    assert d == 1 + 2 * 2
Exemplo n.º 2
0
def test_cartesianvector_dist_cubic():
    v = CartesianVector(1, -4.5)

    d = v.dist_cubic()
    assert d == 1 + 4.5

    v2 = CartesianVector(1, -3.5)
    d = v.dist_cubic(v2)
    assert d == 1
Exemplo n.º 3
0
def test_cartesianvector_floor():
    v1 = CartesianVector(1.5, -4.5)

    v2 = v1.floor()

    assert v2 is not v1

    assert len(v2) == 2
    assert v2[0] == 1
    assert v2[1] == -5
Exemplo n.º 4
0
def test_cartesianvector_ifloor():
    v1 = CartesianVector(1.5, -4.5)

    vbackup = v1
    v1.ifloor()

    assert v1 is vbackup

    assert len(v1) == 2
    assert v1[0] == 1
    assert v1[1] == -5
Exemplo n.º 5
0
def test_cartesianvector_sub():
    v1 = CartesianVector(-1, 2, 5)
    v2 = CartesianVector(5, 4, 3)
    v3 = v1 - v2

    # make sure they are not the same object
    assert v3 is not v1
    assert v3 is not v2

    assert len(v3) == 3
    assert v3[0] == -6
    assert v3[1] == -2
    assert v3[2] == 2
Exemplo n.º 6
0
def test_cartesianvector_isub():
    v1 = CartesianVector(-1, 2, 5)
    v2 = CartesianVector(5, 4, 3)

    vbackup = v1
    v1 -= v2

    # make sure same object
    assert v1 is vbackup
    assert v1 is not v2

    assert len(v1) == 3
    assert v1[0] == -6
    assert v1[1] == -2
    assert v1[2] == 2
Exemplo n.º 7
0
def test_cartesianvector_mul():
    v1 = CartesianVector(1, 2)

    v2 = v1 * 2

    assert v2 is not v1

    assert len(v2) == 2
    assert v2[0] == 2
    assert v2[1] == 4
Exemplo n.º 8
0
def test_cartesianvector_imul():
    v1 = CartesianVector(1, 2)

    vbackup = v1
    v1 *= 2

    assert v1 is vbackup

    assert len(v1) == 2
    assert v1[0] == 2
    assert v1[1] == 4
Exemplo n.º 9
0
def test_cartesianvector_div():
    v1 = CartesianVector(1, 4)

    v2 = v1 / 2

    assert v2 is not v1

    assert len(v2) == 2
    # because we 'from __future__ import division' in vector.py
    # py2 and py3 should be the same
    assert v2[0] == 0.5
    assert v2[1] == 2.0
Exemplo n.º 10
0
def test_cartesianvector_add():
    v1 = CartesianVector(-1, 2, 1)

    v2 = v1 + v1

    # make sure they are not the same object
    assert v2 is not v1

    assert len(v2) == 3
    assert v2[0] == -2
    assert v2[1] == 4
    assert v2[2] == 2
Exemplo n.º 11
0
def test_cartesianvector_iadd():
    v1 = CartesianVector(-1, 2, 1)

    vbackup = v1
    v1 += v1

    # make sure same object
    assert v1 is vbackup

    assert len(v1) == 3
    assert v1[0] == -2
    assert v1[1] == 4
    assert v1[2] == 2
Exemplo n.º 12
0
def test_cartesianvector_idiv():
    v1 = CartesianVector(1, 4)

    vbackup = v1
    v1 /= 2

    assert v1 is vbackup

    assert len(v1) == 2
    # because we 'from __future__ import division' in vector.py
    # py2 and py3 should be the same
    assert v1[0] == 0.5
    assert v1[1] == 2.0