def test_mkdir(self, mock_stdout):
# test the dit has exists
with mock.patch('main.os.path.exists', return_value=True):
self.assertEqual(main.mkdir(BACKUP_DIR), self.path)
main.os.path.join.assert_called_with(BASE_DIR, BACKUP_DIR, 'time')
main.os.makedirs.assert_not_called()
self.assertTrue(mock_stdout.getvalue() == 'doc has existed\n')
# test the dir has not exists
with mock.patch('main.os.path.exists', return_value=False):
self.assertEqual(main.mkdir(BACKUP_DIR), self.path)
main.os.makedirs.assert_called_with(self.path)
self.assertEqual(main.os.makedirs.call_count, len(DIRNAMES) - 1)
def test_mkdir_false(self):
target = runtime + 'test_mkdir/'
expected = False
if not os.path.exists(target):
os.makedirs(target)
self.assertEqual(
main.mkdir(target), expected, 'test [mkdir_false] fail!')
def test_mkdir_true(self):
target = runtime + 'test_mkdir/'
expected = True
if os.path.exists(target):
os.removedirs(target)
self.assertEqual(
main.mkdir(target), expected, 'test [mkdir_true] fail!')
def handle_all_media(message):
print("Recibiendo")
file_id = message.json["photo"][0]["file_id"]
file_info = bot.get_file(file_id)
print(file_info.file_path)
file = requests.get('https://api.telegram.org/file/bot{0}/{1}'.format(
API_TOKEN, file_info.file_path))
print(file)
main.mkdir()
if file.status_code == 200:
with open("images/imagen.jpg", "wb") as f:
f.write(file.content)
if main.lengh('images/imagen.jpg') > 3:
bot.reply_to(message,
"Texto: \n" + main.convertidor('images/imagen.jpg'))
else:
bot.reply_to(message, "No se reconoce la imagen :( \nprueba con otra.")
from all_albums import get_all_albums
from main import mkdir,download_pic,all_num
import time
if __name__ == '__main__':
get_all_albums() # 获取所有板块下所有专辑的链接，保存至txt文本
time.sleep(1)
mkdir() # 创建所有板块图片的主目录
time.sleep(1)
download_pic() # 开始抓取图片，目录层级自动创建
all_num() # 统计一共抓取了多少专辑和图片
if os.path.exists(target):
os.removedirs(target)
self.assertEqual(
main.mkdir(target), expected, 'test [mkdir_true] fail!')
def test_mkdir_false(self):
target = runtime + 'test_mkdir/'
expected = False
if not os.path.exists(target):
os.makedirs(target)
self.assertEqual(
main.mkdir(target), expected, 'test [mkdir_false] fail!')
if __name__ == '__main__':
main.mkdir(runtime)
import xmlrunner
#unittest.main(module=__name__, buffer=True, verbosity=2, exit=False)
with open(runtime + 'test-results.xml', 'wb') as output:
unittest.main(
testRunner=xmlrunner.XMLTestRunner(output=output),
module=__name__,
buffer=True,
verbosity=2,
exit=False)
# vim: noai:ts=4:sw=4
args.outpath = os.path.join(args.outbasepath,args.modelname,str(args.runId).zfill(2)) args.save_dir = os.path.join(args.outpath, "save") args.save_dir_best = os.path.join(args.outpath, "save-best") if args.load_best: args.load = True if args.load_from is not None: args.load = True if args.mode == "train": assert args.outbasepath is not None assert args.modelname is not None args.is_train = True mkdir(args.save_dir) mkdir(args.save_dir_best) else: args.is_train = False args.num_epochs = 1 args.resnet_num_block = [3, 4, 23, 3] # resnet 101 args.use_basic_block = False if args.resnet152: args.resnet_num_block = [3, 8, 36, 3] if args.resnet50: args.resnet_num_block = [3, 4, 6, 3] if args.resnet34: args.resnet_num_block = [3, 4, 6, 3] args.use_basic_block = True if args.resnet18: