44import pytest
55from builtins import *
66
7- import canmatrix .canmatrix
7+ from canmatrix .CanMatrix import CanMatrix
8+ from canmatrix .Signal import Signal
9+ from canmatrix .Frame import Frame
10+ from canmatrix .ArbitrationId import ArbitrationId
11+ from canmatrix .Ecu import Ecu
12+
13+ @pytest .fixture
14+ def empty_frame ():
15+ return Frame (name = "test_frame" )
816
917
1018def test_signal_defaults_to_decimal ():
11- signal = canmatrix . canmatrix . Signal (
19+ signal = Signal (
1220 offset = 4 ,
1321 factor = 2 ,
1422 )
@@ -910,10 +918,10 @@ def test_arbitration_id_j1939_direct_setters():
910918 assert arb_id .j1939_priority == 3
911919
912920def test_arbitration_id_comparators ():
913- id_standard_1 = canmatrix . ArbitrationId (id = 0x1 , extended = False )
914- id_standard_2 = canmatrix . ArbitrationId (id = 0x2 , extended = False )
915- id_extended_1 = canmatrix . ArbitrationId (id = 0x1 , extended = True )
916- id_extended_2 = canmatrix . ArbitrationId (id = 0x2 , extended = True )
921+ id_standard_1 = ArbitrationId (id = 0x1 , extended = False )
922+ id_standard_2 = ArbitrationId (id = 0x2 , extended = False )
923+ id_extended_1 = ArbitrationId (id = 0x1 , extended = True )
924+ id_extended_2 = ArbitrationId (id = 0x2 , extended = True )
917925
918926 sorting_results = sorted ((
919927 id_extended_1 , id_standard_2 , id_extended_2 , id_standard_1 ))
@@ -924,7 +932,7 @@ def test_arbitration_id_comparators():
924932
925933@pytest .fixture
926934def empty_matrix ():
927- return canmatrix . CanMatrix ()
935+ return CanMatrix ()
928936
929937
930938def test_canmatrix_add_attribure (empty_matrix ):
@@ -934,7 +942,7 @@ def test_canmatrix_add_attribure(empty_matrix):
934942
935943def test_canmatrix_get_frame_by_glob (empty_matrix , empty_frame ):
936944 empty_matrix .add_frame (empty_frame )
937- f2 = canmatrix . Frame (name = "nm_osek_esp" )
945+ f2 = Frame (name = "nm_osek_esp" )
938946 empty_matrix .add_frame (f2 )
939947 assert empty_matrix .glob_frames ("*osek*" ) == [f2 ]
940948
@@ -948,6 +956,7 @@ def test_canmatrix_get_frame_by_wrong_name(empty_matrix, empty_frame):
948956 empty_matrix .add_frame (empty_frame )
949957 assert empty_matrix .frame_by_name ("wrong" ) is None
950958
959+
951960def test_canmatrix_get_frame_by_pgn (empty_matrix , empty_frame ):
952961 empty_frame .arbitration_id .id = 0xA123456
953962 empty_frame .arbitration_id .extended = True
@@ -968,29 +977,29 @@ def test_canmatrix_iterate_over_frames(empty_matrix, empty_frame):
968977
969978def test_canmatrix_remove_frame (empty_matrix , empty_frame ):
970979 empty_matrix .add_frame (empty_frame )
971- empty_matrix .add_frame (canmatrix . Frame ())
980+ empty_matrix .add_frame (Frame ())
972981 empty_matrix .remove_frame (empty_frame )
973982 assert len (empty_matrix .frames ) == 1
974983
975984
976985def test_canmatrix_rename_ecu_by_name (empty_matrix ):
977- ecu = canmatrix . Ecu (name = "old_name" )
986+ ecu = Ecu (name = "old_name" )
978987 empty_matrix .add_ecu (ecu )
979988 empty_matrix .rename_ecu ("old_name" , "new name" )
980989 assert ecu .name == "new name"
981990
982991
983992def test_canmatrix_rename_ecu_by_wrong_name (empty_matrix ):
984- ecu = canmatrix . Ecu (name = "old_name" )
993+ ecu = Ecu (name = "old_name" )
985994 empty_matrix .add_ecu (ecu )
986995 empty_matrix .rename_ecu ("wrong" , "new name" )
987996 assert ecu .name == "old_name"
988997
989998
990999def test_canmatrix_rename_ecu_check_frame (empty_matrix ):
991- ecu = canmatrix . Ecu (name = "old_name" )
992- frame = canmatrix . Frame (name = "test_frame" )
993- signal = canmatrix . Signal (name = "test_signal" )
1000+ ecu = Ecu (name = "old_name" )
1001+ frame = Frame (name = "test_frame" )
1002+ signal = Signal (name = "test_signal" )
9941003 signal .add_receiver ("old_name" )
9951004 frame .add_signal (signal )
9961005 frame .update_receiver ()
@@ -1003,19 +1012,19 @@ def test_canmatrix_rename_ecu_check_frame(empty_matrix):
10031012 assert "new_name" in frame .receivers
10041013
10051014def test_canmatrix_rename_ecu_by_instance (empty_matrix ):
1006- ecu = canmatrix . Ecu (name = "old_name" )
1015+ ecu = Ecu (name = "old_name" )
10071016 empty_matrix .add_ecu (ecu )
10081017 empty_matrix .rename_ecu (ecu , "new name" )
10091018 assert ecu .name == "new name"
10101019
10111020
10121021def test_canmatrix_del_ecu_by_glob (empty_matrix ):
1013- ecu1 = canmatrix . Ecu (name = "ecu1" )
1014- ecu2 = canmatrix . Ecu (name = "ecu2" )
1015- frame = canmatrix . Frame (transmitters = ["ecu2" , "ecu3" ])
1022+ ecu1 = Ecu (name = "ecu1" )
1023+ ecu2 = Ecu (name = "ecu2" )
1024+ frame = Frame (transmitters = ["ecu2" , "ecu3" ])
10161025 empty_matrix .add_ecu (ecu1 )
10171026 empty_matrix .add_ecu (ecu2 )
1018- frame .add_signal (canmatrix . Signal (receivers = ["ecu1" , "ecu2" ]))
1027+ frame .add_signal (Signal (receivers = ["ecu1" , "ecu2" ]))
10191028 empty_matrix .add_frame (frame )
10201029 empty_matrix .del_ecu ("*2" )
10211030 assert empty_matrix .ecus == [ecu1 ]
@@ -1024,19 +1033,19 @@ def test_canmatrix_del_ecu_by_glob(empty_matrix):
10241033
10251034
10261035def test_canmatrix_del_ecu_by_instance (empty_matrix ):
1027- ecu1 = canmatrix . Ecu (name = "ecu1" )
1028- ecu2 = canmatrix . Ecu (name = "ecu2" )
1036+ ecu1 = Ecu (name = "ecu1" )
1037+ ecu2 = Ecu (name = "ecu2" )
10291038 empty_matrix .add_ecu (ecu1 )
10301039 empty_matrix .add_ecu (ecu2 )
10311040 empty_matrix .del_ecu (ecu1 )
10321041 assert empty_matrix .ecus == [ecu2 ]
10331042
10341043
10351044def test_canmatrix_del_obsolete_ecus (empty_matrix ):
1036- empty_matrix .add_ecu (canmatrix . Ecu (name = "Ecu1" ))
1037- empty_matrix .add_ecu (canmatrix . Ecu (name = "Ecu2" ))
1038- frame1 = canmatrix . Frame (name = "frame1" , transmitters = ["Ecu1" ])
1039- frame1 .add_signal (canmatrix . Signal ("signal1" , receivers = ["Ecu2" ]))
1045+ empty_matrix .add_ecu (Ecu (name = "Ecu1" ))
1046+ empty_matrix .add_ecu (Ecu (name = "Ecu2" ))
1047+ frame1 = Frame (name = "frame1" , transmitters = ["Ecu1" ])
1048+ frame1 .add_signal (Signal ("signal1" , receivers = ["Ecu2" ]))
10401049 empty_matrix .add_frame (frame1 )
10411050 empty_matrix .delete_obsolete_ecus ()
10421051 assert "Ecu1" in [ecu .name for ecu in empty_matrix .ecus ]
@@ -1048,7 +1057,7 @@ def test_canmatrix_del_obsolete_ecus(empty_matrix):
10481057
10491058
10501059def test_canmatrix_rename_frame_by_name (empty_matrix ):
1051- f = canmatrix . Frame (name = "F1" )
1060+ f = Frame (name = "F1" )
10521061 empty_matrix .add_frame (f )
10531062 empty_matrix .rename_frame ("F1" , "F2" )
10541063 assert f .name == "F2"
@@ -1063,15 +1072,15 @@ def test_canmatrix_rename_frame_by_name(empty_matrix):
10631072
10641073
10651074def test_canmatrix_rename_frame_by_instance (empty_matrix ):
1066- f = canmatrix . Frame (name = "F1" )
1075+ f = Frame (name = "F1" )
10671076 empty_matrix .add_frame (f )
10681077 empty_matrix .rename_frame (f , "F2" )
10691078 assert f .name == "F2"
10701079
10711080
10721081def test_canmatrix_del_frame_by_name (empty_matrix ):
1073- f1 = canmatrix . Frame (name = "F1" )
1074- f2 = canmatrix . Frame (name = "F2" )
1082+ f1 = Frame (name = "F1" )
1083+ f2 = Frame (name = "F2" )
10751084 empty_matrix .add_frame (f1 )
10761085 empty_matrix .add_frame (f2 )
10771086 empty_matrix .del_frame ("F1" )
@@ -1080,17 +1089,17 @@ def test_canmatrix_del_frame_by_name(empty_matrix):
10801089
10811090
10821091def test_canmatrix_del_frame_by_instance (empty_matrix ):
1083- f1 = canmatrix . Frame (name = "F1" )
1084- f2 = canmatrix . Frame (name = "F2" )
1092+ f1 = Frame (name = "F1" )
1093+ f2 = Frame (name = "F2" )
10851094 empty_matrix .add_frame (f1 )
10861095 empty_matrix .add_frame (f2 )
10871096 empty_matrix .del_frame (f1 )
10881097 assert empty_matrix .frames == [f2 ]
10891098
10901099def test_effective_cycle_time ():
1091- frame = canmatrix . Frame ()
1092- sig1 = canmatrix . Signal (name = "s1" , cycle_time = 1 )
1093- sig2 = canmatrix . Signal (name = "s2" , cycle_time = 0 )
1100+ frame = Frame ()
1101+ sig1 = Signal (name = "s1" , cycle_time = 1 )
1102+ sig2 = Signal (name = "s2" , cycle_time = 0 )
10941103 frame .add_signal (sig1 )
10951104 frame .add_signal (sig2 )
10961105 assert frame .effective_cycle_time == 1
@@ -1113,30 +1122,30 @@ def test_effective_cycle_time():
11131122 assert frame .effective_cycle_time == 0
11141123
11151124def test_baudrate ():
1116- cm = canmatrix . CanMatrix ()
1125+ cm = CanMatrix ()
11171126 cm .baudrate = 500000
11181127 assert cm .baudrate == 500000
11191128 cm .fd_baudrate = 1000000
11201129 assert cm .fd_baudrate == 1000000
11211130
11221131def test_frame_compress ():
1123- frame = canmatrix . Frame ("my_frame" , size = 8 )
1124- frame .add_signal (canmatrix . Signal (name = "Sig1" , start_bit = 2 , size = 13 , is_little_endian = False ))
1125- frame .add_signal (canmatrix . Signal (name = "Sig2" , start_bit = 17 , size = 14 , is_little_endian = False ))
1126- frame .add_signal (canmatrix . Signal (name = "Sig3" , start_bit = 35 , size = 6 , is_little_endian = False ))
1127- frame .add_signal (canmatrix . Signal (name = "Sig4" , start_bit = 49 , size = 8 , is_little_endian = False ))
1132+ frame = Frame ("my_frame" , size = 8 )
1133+ frame .add_signal (Signal (name = "Sig1" , start_bit = 2 , size = 13 , is_little_endian = False ))
1134+ frame .add_signal (Signal (name = "Sig2" , start_bit = 17 , size = 14 , is_little_endian = False ))
1135+ frame .add_signal (Signal (name = "Sig3" , start_bit = 35 , size = 6 , is_little_endian = False ))
1136+ frame .add_signal (Signal (name = "Sig4" , start_bit = 49 , size = 8 , is_little_endian = False ))
11281137 frame .compress ()
11291138 assert frame .signal_by_name ("Sig1" ).start_bit == 0
11301139 assert frame .signal_by_name ("Sig2" ).start_bit == 13
11311140 assert frame .signal_by_name ("Sig3" ).start_bit == 27
11321141 assert frame .signal_by_name ("Sig4" ).start_bit == 33
11331142
1134- frame = canmatrix . Frame ("my_frame" , size = 8 )
1143+ frame = Frame ("my_frame" , size = 8 )
11351144 # some signals overlap!
1136- frame .add_signal (canmatrix . Signal (name = "Sig1" , start_bit = 12 , size = 12 , is_little_endian = True ))
1137- frame .add_signal (canmatrix . Signal (name = "Sig2" , start_bit = 17 , size = 9 , is_little_endian = True ))
1138- frame .add_signal (canmatrix . Signal (name = "Sig3" , start_bit = 33 , size = 5 , is_little_endian = True ))
1139- frame .add_signal (canmatrix . Signal (name = "Sig4" , start_bit = 48 , size = 9 , is_little_endian = True ))
1145+ frame .add_signal (Signal (name = "Sig1" , start_bit = 12 , size = 12 , is_little_endian = True ))
1146+ frame .add_signal (Signal (name = "Sig2" , start_bit = 17 , size = 9 , is_little_endian = True ))
1147+ frame .add_signal (Signal (name = "Sig3" , start_bit = 33 , size = 5 , is_little_endian = True ))
1148+ frame .add_signal (Signal (name = "Sig4" , start_bit = 48 , size = 9 , is_little_endian = True ))
11401149 frame .compress ()
11411150 assert frame .signal_by_name ("Sig1" ).start_bit == 0
11421151 assert frame .signal_by_name ("Sig2" ).start_bit == 12
0 commit comments