class TestAtCoderClientReal(unittest.TestCase):
def setUp(self):
self.temp_dir = tempfile.mkdtemp()
self.client = AtCoderClient()
def test_submit_source_code(self):
problem_list = self.client.download_problem_list(Contest("arc002"))
self.assertEqual(
['arc002_1',
'arc002_2',
'arc002_3',
'arc002_4'],
[p.problem_id for p in problem_list])
def test_download_problem_content(self):
content = self.client.download_problem_content(
Problem(Contest("arc002"), "C", "arc002_3"))
self.assertEqual("N\nc_{1}c_{2}...c_{N}\n", content.input_format_text)
self.assertEqual(3, len(content.samples))
def test_login_failed(self):
def fake_supplier():
return "@@@ invalid user name @@@", "@@@ password @@@"
try:
self.client.login(credential_supplier=fake_supplier,
use_local_session_cache=False)
self.fail("Unexpectedly, this test succeeded to login.")
except LoginError:
pass
def test_download_all_contests(self):
contests = self.client.download_all_contests()
# Check if the number of contests is more than the number when I wrote
# this unit test.
self.assertGreaterEqual(len(contests), 523)
# Make sure there is no duplication
self.assertEqual(
len(set([c.get_id() for c in contests])),
len(contests))
def test_check_logging_in_is_false(self):
self.assertFalse(self.client.check_logging_in())
def test_cookie_save_and_load(self):
cookie_path = os.path.join(self.temp_dir, "cookie.txt")
session = requests.Session()
loaded = load_cookie_to(session, cookie_path)
self.assertFalse(loaded)
save_cookie(session, cookie_path)
new_session = requests.Session()
loaded = load_cookie_to(new_session, cookie_path)
self.assertTrue(loaded)
atcoder = AtCoderClient()
def mkdirs(path):
try:
os.makedirs(path)
except OSError as exc:
if exc.errno == errno.EEXIST and os.path.isdir(path):
pass
if __name__ == "__main__":
htmls_dir = "./problem_htmls/"
mkdirs(htmls_dir)
for contest in atcoder.download_all_contests():
for problem in atcoder.download_problem_list(contest):
html_path = os.path.join(
htmls_dir, "{contest}-{problem_id}.html".format(
contest=contest.get_id(),
problem_id=problem.get_alphabet()))
if os.path.exists(html_path):
print(
"{} already exists -- skipping download".format(html_path))
continue
print("Downloading {}".format(html_path))
try:
html = atcoder.download_problem_content(problem).original_html
with open(html_path, 'w') as f:
atcoder = AtCoderClient()
def mkdirs(path):
try:
os.makedirs(path)
except OSError as exc:
if exc.errno == errno.EEXIST and os.path.isdir(path):
pass
if __name__ == "__main__":
htmls_dir = "./problem_htmls/"
mkdirs(htmls_dir)
for contest in atcoder.download_all_contests():
for problem in atcoder.download_problem_list(contest):
html_path = os.path.join(htmls_dir, "{contest}-{problem_id}.html".format(
contest=contest.get_id(), problem_id=problem.get_alphabet()))
if os.path.exists(html_path):
print(
"{} already exists -- skipping download".format(html_path))
continue
print("Downloading {}".format(html_path))
try:
html = atcoder.download_problem_content(problem).original_html
with open(html_path, 'w') as f:
f.write(html)
except Exception as e:
