from src.compute import project_ortho, project_perspective

def test_ortho_proj_unscaled():
    #Arrange
    point = (1.0, 2.0, 3.0)
    expected = (1.0, 2.0)

    #Act
    result = project_ortho(point)

    #Assert
    assert result == expected


def test_ortho_proj_scaled():
    # Arrange
    point = (1.0, 2.0, 3.0)
    scale = 100.0
    expected = (100.0, 200.0)

    # Act
    result = project_ortho(point, scale)

    # Assert
    assert result == expected

def test_perspective_proj_unscaled():
    # Arrange
    point = (1.0, 2.0, 3.0)
    expected = (1/7, 2/7)

    # Act
    result = project_perspective(point)

    # Assert
    assert result == expected

def test_perspective_proj_scaled():
    # Arrange
    point = (1.0, 2.0, 3.0)
    scale = 100.0
    focal_length = 2
    expected = (200/7, 400/7)

    # Act
    result = project_perspective(point, scale, focal_length)

    # Assert
    assert result == expected