def test_vlans_length(self): result_1 = main_script_functions.vlans_df_primary_port_sorter( 'test_vlans.csv', 0) result_2 = main_script_functions.vlans_df_primary_port_sorter( 'test_vlans.csv', 1) combined_length = len(result_1) + len(result_2) length_input = len(main_script_functions.load_file('test_vlans.csv')) self.assertEqual(combined_length, length_input)
def test_requests_length(self): result_1 = main_script_functions.redundant_df_loader_sorter( 'test_requests.csv', 0) result_2 = main_script_functions.redundant_df_loader_sorter( 'test_requests.csv', 1) combined_length = len(result_1) + len(result_2) length_input = len( main_script_functions.load_file('test_requests.csv')) self.assertEqual(combined_length, length_input)
def test_final_df_length(self): length_1 = len(main_script_functions.load_file('test_requests.csv')) length_non_redundant = len( main_script_functions.redundant_df_loader_sorter( 'test_requests.csv', 1)) redundant_df = main_script_functions.redundant_df_loader_sorter( 'test_requests.csv', 1) non_redundant_df = main_script_functions.redundant_df_loader_sorter( 'test_requests.csv', 0) vlans_df_primary_port = main_script_functions.vlans_df_primary_port_sorter( 'test_vlans.csv', 1) vlans_df_primary_port_used = main_script_functions.vlans_df_primary_port_sorter( 'test_vlans.csv', 1) vlans_df_non_primary_port = main_script_functions.vlans_df_primary_port_sorter( 'test_vlans.csv', 0) #for i in range(0,10): json_redundant_df = json.loads(redundant_df.to_json(orient='records')) json_non_redundant_df = json.loads( non_redundant_df.to_json(orient='records')) final_output_redundant_parser_list = main_script_functions.redundant_df_parser( json_redundant_df, vlans_df_primary_port, vlans_df_primary_port_used, vlans_df_non_primary_port) vlans_df_primary_port = final_output_redundant_parser_list[0] vlans_df_primary_port_used = final_output_redundant_parser_list[1] vlans_df_non_primary_port = final_output_redundant_parser_list[2] final_redundant_df = final_output_redundant_parser_list[3] final_non_redundant_df_list = main_script_functions.non_redundant_df_parser( json_non_redundant_df, vlans_df_primary_port, vlans_df_primary_port_used, vlans_df_non_primary_port) vlans_df_primary_port = final_non_redundant_df_list[0] vlans_df_primary_port_used = final_non_redundant_df_list[1] vlans_df_non_primary_port = final_non_redundant_df_list[2] final_non_redundant_df = final_non_redundant_df_list[3] final_output_df = final_redundant_df.append( final_non_redundant_df, ignore_index=True).sort_values( ['request_id', 'primary_port', 'device_id'], ascending=[True, True, True]).reset_index(drop=True).drop( ['redundant'], axis=1)[[ 'request_id', 'device_id', 'primary_port', 'vlan_id' ]] print(final_output_df.head(10)) expected_length = length_1 + length_non_redundant length_2 = len(final_output_df) print("length of the input dataframe is : " + str(length_1)) print("length of the redundant dataframe is : " + str(length_non_redundant)) print("length of the final output dataframe is : " + str(length_2)) self.assertEqual(expected_length, length_2)
def test_final_df_values(self): redundant_df = main_script_functions.redundant_df_loader_sorter( 'test_requests.csv', 1) non_redundant_df = main_script_functions.redundant_df_loader_sorter( 'test_requests.csv', 0) vlans_df_primary_port = main_script_functions.vlans_df_primary_port_sorter( 'test_vlans.csv', 1) vlans_df_primary_port_used = main_script_functions.vlans_df_primary_port_sorter( 'test_vlans.csv', 1) vlans_df_non_primary_port = main_script_functions.vlans_df_primary_port_sorter( 'test_vlans.csv', 0) #for i in range(0,10): json_redundant_df = json.loads(redundant_df.to_json(orient='records')) json_non_redundant_df = json.loads( non_redundant_df.to_json(orient='records')) final_output_redundant_parser_list = main_script_functions.redundant_df_parser( json_redundant_df, vlans_df_primary_port, vlans_df_primary_port_used, vlans_df_non_primary_port) vlans_df_primary_port = final_output_redundant_parser_list[0] vlans_df_primary_port_used = final_output_redundant_parser_list[1] vlans_df_non_primary_port = final_output_redundant_parser_list[2] final_redundant_df = final_output_redundant_parser_list[3] final_non_redundant_df_list = main_script_functions.non_redundant_df_parser( json_non_redundant_df, vlans_df_primary_port, vlans_df_primary_port_used, vlans_df_non_primary_port) vlans_df_primary_port = final_non_redundant_df_list[0] vlans_df_primary_port_used = final_non_redundant_df_list[1] vlans_df_non_primary_port = final_non_redundant_df_list[2] final_non_redundant_df = final_non_redundant_df_list[3] final_output_df = final_redundant_df.append( final_non_redundant_df, ignore_index=True).sort_values( ['request_id', 'primary_port', 'device_id'], ascending=[True, True, True]).reset_index(drop=True).drop( ['redundant'], axis=1)[[ 'request_id', 'device_id', 'primary_port', 'vlan_id' ]] test_output_df = main_script_functions.load_file('test_output.csv') dataframe_test_value = test_output_df.equals(final_output_df) print("test output dataframe is: \n") print(test_output_df) print("actual output dataframe is: \n") print(final_output_df) self.assertEqual(dataframe_test_value, True)
def test_df_requests_loader(self): result = main_script_functions.load_file('test_requests.csv') test_result = len(result) self.assertNotEqual(test_result, 0)