def test_prediction_return_type(self):
check_none = pd.prediction(pnd.read_csv(self.path),
['Temperature', 'Error'], "Failure", 1)
self.assertIsNone(check_none)
check_none = pd.prediction(pnd.read_csv(self.path), ['Temperature'],
"Failure", 2)
self.assertIsNotNone(check_none)
self.assertEqual(type(check_none), str)
def test_save_model_file(self):
home = str(Path.home())
count_file_after = 0
list_dir = os.listdir(home)
for file in list_dir:
if file.endswith('.pkl'):
os.remove(home + "\\" + file)
pd.prediction(pnd.read_csv(self.path), ['Temperature'], "Failure", 1)
for file in list_dir:
if file.endswith('.pkl'):
count_file_after += 1
self.assertTrue(count_file_after > 0)
def init_result_window(self):
self.title("Result prediction window")
self.geometry("400x240+300+200")
self.focus_get()
self.resizable(False, False)
self.var.set(0)
result_label = Label(self, text="Result prediction", font="Arial 15", pady=15, padx=120)
result_label.grid(row=0, column=0, sticky=W)
btn_back = tkinter.Button(self, text="Back", command=self.exit_window, anchor=SW, padx=10)
btn_back.place(x=3, y=210)
success_label = Label(text="\nResult was written\nin file prediction_data_<algorithm>")
text = Text(self, width=400, height=8)
text.insert(1.0, pd.prediction(self.data, self.params, self.fail_col_name, self.algorithm))
text.tag_config('info', foreground="green")
text.insert(END, success_label['text'], 'info')
text.configure(state=DISABLED)
text.grid(row=1, column=0, sticky=W)
home = str(Path.home())
if self.algorithm == 2:
dw.DataWindow(self.root, home + "\\prediction_data_lr.csv")
else:
dw.DataWindow(self.root, home + "\\prediction_data_rf.csv")
def prediction(self):
params = []
for el in self.lst:
params.append(el.get())
try:
flag = pd.prediction(dpm.read_file_prediction(self.file), params,
self.fail_col_name, self.algorithm)
if flag is None:
raise Exception("Please, enter correct name of column")
rpw.ResultPredictionWindow(self,
dpm.read_file_prediction(self.file),
params, self.fail_col_name,
self.algorithm, self.file)
self.withdraw()
except Exception as e:
messagebox.showerror(title="Error", message=e)
def test_out_predict_data_save_file(self):
home = str(Path.home())
pd.prediction(pnd.read_csv(self.path), ['Temperature'], "Failure", 1)
pd.prediction(pnd.read_csv(self.path), ['Temperature'], "Failure", 2)
self.assertTrue(os.path.exists(home + "\\prediction_data_lr.csv"))
self.assertTrue(os.path.exists(home + "\\prediction_data_rf.csv"))