CopterExpress/clever-show
1
0
Fork 0
mirror of https://github.com/CopterExpress/clever-show.git synced 2026-05-31 01:09:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: Fix animation module test

Browse Source
This commit is contained in:
Arthur Golubtsov
2020-06-03 12:24:34 +03:00
parent 500ffa0795
commit 09f32469f4
1 changed files with 47 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
test/animation_test.py
View File

@@ -4,12 +4,12 @@ import shutil
from pytest import approx
import pytest
# Add parent dir to PATH to import config
import inspect
current_dir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
parent_dir = os.path.dirname(current_dir)
sys.path.insert(0, parent_dir)
sys.path.insert(0, os.path.join(parent_dir,"Drone"))
# Add parent dir to PATH to import messaging_lib and config_lib
current_dir = (os.path.dirname(os.path.realpath(__file__)))
lib_dir = os.path.realpath(os.path.join(current_dir, '../lib'))
drone_dir = os.path.realpath(os.path.join(current_dir, '../drone'))
sys.path.insert(0, lib_dir)
sys.path.insert(0, drone_dir)
from config import ConfigManager
@@ -23,29 +23,29 @@ if not os.path.exists(spec_path):
else:
print("Successfully created the directory {}".format(spec_path))
shutil.copy("../Drone/config/spec/configspec_client.ini", spec_path)
shutil.copy("../drone/config/spec/configspec_client.ini", spec_path)
config = ConfigManager()
config.load_config_and_spec(os.path.join(config_path,'client.ini'))
assert config.config_name == "client"
import animation_lib
import modules.animation as animation
a = animation_lib.Animation()
a = animation.Animation()
def test_animation_1():
a.update_frames(config, "animation_1.csv")
a.update_frames(config, "assets/animation_1.csv")
assert a.id == 'basic'
assert approx(a.original_frames[0].get_pos()) == [0,0,0]
assert a.original_frames[0].get_color() == [204,2,0]
assert a.original_frames[0].pose_is_valid()
assert animation_lib.get_numbers(a.static_begin_frames) == range(1,11)
assert animation_lib.get_numbers(a.takeoff_frames) == range(11,21)
assert animation_lib.get_numbers(a.route_frames) == range(21,31)
assert animation_lib.get_numbers(a.land_frames) == range(31, 41)
assert animation_lib.get_numbers(a.static_end_frames) == range(41, 51)
assert animation_lib.get_numbers(a.output_frames) == range(11,31)
assert animation.get_numbers(a.static_begin_frames) == range(1,11)
assert animation.get_numbers(a.takeoff_frames) == range(11,21)
assert animation.get_numbers(a.route_frames) == range(21,31)
assert animation.get_numbers(a.land_frames) == range(31, 41)
assert animation.get_numbers(a.static_end_frames) == range(41, 51)
assert animation.get_numbers(a.output_frames) == range(11,31)
assert approx(a.static_begin_time) == 1
assert approx(a.takeoff_time) == 1
assert approx(a.output_frames_min_z) == 0.1
@@ -54,17 +54,17 @@ def test_animation_1():
assert approx(a.get_start_point(ratio=[1,2,3], offset=[4,5,6])) == [4.,5.,6.3]
def test_animation_2():
a.update_frames(config, "animation_2.csv")
a.update_frames(config, "assets/animation_2.csv")
assert a.id == 'parad'
assert approx(a.original_frames[271].get_pos()) == [-1.00519,2.65699,0.24386]
assert a.original_frames[271].get_color() == [7,255,0]
assert a.original_frames[271].pose_is_valid()
assert animation_lib.get_numbers(a.static_begin_frames) == range(271)
assert animation_lib.get_numbers(a.takeoff_frames) == range(271,285)
assert animation_lib.get_numbers(a.route_frames) == range(285,1065)
assert animation_lib.get_numbers(a.land_frames) == []
assert animation_lib.get_numbers(a.static_end_frames) == []
assert animation_lib.get_numbers(a.output_frames) == range(271, 1065)
assert animation.get_numbers(a.static_begin_frames) == range(271)
assert animation.get_numbers(a.takeoff_frames) == range(271,285)
assert animation.get_numbers(a.route_frames) == range(285,1065)
assert animation.get_numbers(a.land_frames) == []
assert animation.get_numbers(a.static_end_frames) == []
assert animation.get_numbers(a.output_frames) == range(271, 1065)
assert approx(a.static_begin_time) == 27.1
assert approx(a.takeoff_time) == 1.4
assert approx(a.output_frames_min_z) == 0.24386
@@ -73,16 +73,16 @@ def test_animation_2():
assert approx(a.get_start_point(ratio=[1,2,3], offset=[4,5,6])) == [2.99481, 10.31398, 6.73158]
def test_animation_3():
a.update_frames(config, "animation_3.csv")
a.update_frames(config, "assets/animation_3.csv")
assert a.id == 'route'
assert approx(a.original_frames[9].get_pos()) == [0.97783,0.0,1.0]
assert a.original_frames[9].get_color() == [0,204,2]
assert a.original_frames[9].pose_is_valid()
assert animation_lib.get_numbers(a.static_begin_frames) == []
assert animation_lib.get_numbers(a.takeoff_frames) == []
assert animation_lib.get_numbers(a.route_frames) == range(20,31)
assert animation_lib.get_numbers(a.land_frames) == []
assert animation_lib.get_numbers(a.static_end_frames) == []
assert animation.get_numbers(a.static_begin_frames) == []
assert animation.get_numbers(a.takeoff_frames) == []
assert animation.get_numbers(a.route_frames) == range(20,31)
assert animation.get_numbers(a.land_frames) == []
assert animation.get_numbers(a.static_end_frames) == []
assert approx(a.static_begin_time) == 0
assert approx(a.takeoff_time) == 0
assert approx(a.output_frames_min_z) == 1
@@ -91,17 +91,17 @@ def test_animation_3():
assert approx(a.get_start_point(ratio=[1,2,3], offset=[4,5,6])) == [4,5,9]
def test_animation_4():
a.update_frames(config, "animation_4.csv")
a.update_frames(config, "assets/animation_4.csv")
assert a.id == 'two_drones_test'
assert approx(a.original_frames[11].get_pos()) == [0.21774,1.4,1.0]
assert a.original_frames[11].get_color() == [0,0,0]
assert a.original_frames[11].pose_is_valid()
assert animation_lib.get_numbers(a.static_begin_frames) == range(1,12)
assert animation_lib.get_numbers(a.takeoff_frames) == []
assert animation_lib.get_numbers(a.route_frames) == range(12,141)
assert animation_lib.get_numbers(a.land_frames) == []
assert animation_lib.get_numbers(a.static_end_frames) == range(141,161)
assert animation_lib.get_numbers(a.output_frames) == range(12,141)
assert animation.get_numbers(a.static_begin_frames) == range(1,12)
assert animation.get_numbers(a.takeoff_frames) == []
assert animation.get_numbers(a.route_frames) == range(12,141)
assert animation.get_numbers(a.land_frames) == []
assert animation.get_numbers(a.static_end_frames) == range(141,161)
assert animation.get_numbers(a.output_frames) == range(12,141)
assert approx(a.static_begin_time) == 1.1
assert approx(a.takeoff_time) == 0
assert approx(a.output_frames_min_z) == 1
@@ -114,11 +114,11 @@ def test_animation_no_file():
assert a.id == None
assert a.original_frames == []
assert a.output_frames == []
assert animation_lib.get_numbers(a.static_begin_frames) == []
assert animation_lib.get_numbers(a.takeoff_frames) == []
assert animation_lib.get_numbers(a.route_frames) == []
assert animation_lib.get_numbers(a.land_frames) == []
assert animation_lib.get_numbers(a.static_end_frames) == []
assert animation.get_numbers(a.static_begin_frames) == []
assert animation.get_numbers(a.takeoff_frames) == []
assert animation.get_numbers(a.route_frames) == []
assert animation.get_numbers(a.land_frames) == []
assert animation.get_numbers(a.static_end_frames) == []
assert a.static_begin_time == 0
assert a.takeoff_time == 0
assert a.output_frames_min_z is None
@@ -127,12 +127,12 @@ def test_animation_no_file():
assert a.get_start_point(ratio=[1,2,3], offset=[4,5,6]) == []
# print animation_lib.get_numbers(a.static_begin_frames)
# print animation_lib.get_numbers(a.takeoff_frames)
# print animation_lib.get_numbers(a.route_frames)
# print animation_lib.get_numbers(a.land_frames)
# print animation_lib.get_numbers(a.static_end_frames)
# print animation_lib.get_numbers(a.output_frames)
# print animation.get_numbers(a.static_begin_frames)
# print animation.get_numbers(a.takeoff_frames)
# print animation.get_numbers(a.route_frames)
# print animation.get_numbers(a.land_frames)
# print animation.get_numbers(a.static_end_frames)
# print animation.get_numbers(a.output_frames)
# print a.static_begin_time
# print a.takeoff_time
# print a.output_frames_min_z