def test_remove_file(self): path = "/home/" + config["SSH"]["username"] filename = "iris.csv" remotepath = path + "/" + filename ssh.remove_file(remotepath) with self.assertRaises(FileNotFoundError): ssh.import_csv(path, filename)
def test_import_csv(self): path = "/home/" + config["SSH"]["username"] filename = "iris.csv" iris = ssh.import_csv(path, filename) df_len_is_s1 = len(iris) > 10 self.assertEqual(df_len_is_s1, True)