mirror of
https://github.com/louisleroy5/trnslator.git
synced 2024-12-23 08:05:54 -05:00
66 lines
1.5 KiB
Python
66 lines
1.5 KiB
Python
import pytest
|
|
import os
|
|
|
|
import numpy as np
|
|
|
|
from trnslator import utils, timeit, settings
|
|
from geomeppy.geom.polygons import Polygon3D
|
|
from datetime import datetime
|
|
|
|
|
|
def test_rotate(config):
|
|
# Shift list elements to the left
|
|
l1 = [1, 2, 3] # list
|
|
n = 1 # shift 1 position to the left
|
|
l2 = utils.rotate(l1, n)
|
|
|
|
assert l2 == [2, 3, 1]
|
|
|
|
@timeit
|
|
def test_lcm():
|
|
# This function takes two integers and returns the L.C.M.
|
|
x = 10
|
|
y = 50
|
|
lcm = utils.lcm(x, y)
|
|
|
|
assert lcm == 50
|
|
|
|
|
|
def test_float_round(config):
|
|
# Makes sure a variable is a float and round it at "n" decimals
|
|
num = 40.24
|
|
n = 1
|
|
float_num = utils.float_round(num, n)
|
|
|
|
assert float_num == 40.2
|
|
|
|
|
|
def test_angle(config):
|
|
# Calculate the angle between 2 vectors
|
|
# Polygon1 & vector1
|
|
poly1 = Polygon3D(
|
|
[(215.5, 5.0, 0.5), (215.5, 5.0, 2.0), (217.0, 5.0, 2.0), (217.0, 5.0, 0.5)]
|
|
)
|
|
v1 = poly1.normal_vector
|
|
v2 = v1
|
|
angle = utils.angle(v1, v2, acute=False)
|
|
|
|
assert angle == 2 * np.pi
|
|
|
|
|
|
def test_write_lines(config):
|
|
# Delete file if exists, then write lines in it
|
|
path = os.path.join(settings.data_folder, "write_lines.txt")
|
|
lines = ["Test to write lines in file", "2nd line", "end of document"]
|
|
utils.write_lines(path, lines)
|
|
|
|
assert os.path.exists(path)
|
|
|
|
|
|
def test_date_transform(config):
|
|
# Simple function transforming one-based hours (1->24) into zero-based hours (0->23)
|
|
date_str = "08:10"
|
|
new_date = utils.date_transform(date_str)
|
|
|
|
assert new_date == datetime(1900, 1, 1, 7, 10)
|