@@ -61,10 +61,10 @@ def test_fix_break():
6161 test_fix = coordinate_swap .fix_break (broken_mol , 'C2D' , [2.74964 , 2.74964 , 2.74964 ], df_connect_res , False , 1 )
6262
6363 broken_mol_3D = md .load (f'{ input_path } /coord_swap/broken_mol_3D.gro' )
64- test_fix_3D = coordinate_swap .fix_break (broken_mol_3D , 'C2D' , [2.74964 , 2.74964 , 2.74964 ], df_connect_res , False , 1 )
64+ test_fix_3D = coordinate_swap .fix_break (broken_mol_3D , 'C2D' , [2.74964 , 2.74964 , 2.74964 ], df_connect_res , False , 1 ) # noqa: E501
6565
6666 already_fixed = md .load (f'{ input_path } /coord_swap/fixed_mol.gro' )
67- still_fixed = coordinate_swap .fix_break (already_fixed , 'C2D' , [2.74964 , 2.74964 , 2.74964 ], df_connect_res , False , 1 )
67+ still_fixed = coordinate_swap .fix_break (already_fixed , 'C2D' , [2.74964 , 2.74964 , 2.74964 ], df_connect_res , False , 1 ) # noqa: E501
6868
6969 fixed_mol = md .load (f'{ input_path } /coord_swap/fixed_mol.gro' )
7070
@@ -139,8 +139,8 @@ def test_get_miss_coord():
139139
140140 A_dimensions = coordinate_swap .get_dimensions (open (molA_file , 'r' ).readlines ())
141141 B_dimensions = coordinate_swap .get_dimensions (open (molB_file , 'r' ).readlines ())
142- molA = coordinate_swap .fix_break (molA , nameA , A_dimensions , connection_map [connection_map ['Resname' ] == nameA ], False )
143- molB = coordinate_swap .fix_break (molB , nameB , B_dimensions , connection_map [connection_map ['Resname' ] == nameB ], False )
142+ molA = coordinate_swap .fix_break (molA , nameA , A_dimensions , connection_map [connection_map ['Resname' ] == nameA ], False ) # noqa: E501
143+ molB = coordinate_swap .fix_break (molB , nameB , B_dimensions , connection_map [connection_map ['Resname' ] == nameB ], False ) # noqa: E501
144144
145145 df_no_coords = pd .read_csv (f'{ input_path } /coord_swap/extract_missing.csv' )
146146 df = pd .read_csv (f'{ input_path } /coord_swap/df_atom_swap.csv' )
0 commit comments