#201: add a utility to add test cases using --json

tests/download.py

@@ -8,7 +8,22 @@
 import sys
 import tempfile
 
-def snippet_call_download(self, url, files, is_system=False):
+def get_files_from_json(samples):
+    files = {}
+    for i, sample in enumerate(samples):
+        for ext in ('in', 'out'):
+            if 'name' in sample:
+                name = sample['name'] + '.' + ext
+            else:
+                name = 'sample-{}.{}'.format(i + 1, ext)
+            files[name] = sample[ext + 'put']
+    return files
+
+def snippet_call_download(self, url, files, is_system=False, type='files'):
+    assert type in 'files' or 'json'
+    if type == 'json':
+        files = get_files_from_json(files)
+
     cwd = os.getcwd()
     ojtools = os.path.join( cwd, 'oj' )
     try:

tests/download_atcoder.py

@@ -88,3 +88,9 @@ def test_call_download_atcoder_agc029_a(self):
                 'sample-2.in':  '5bdcc020d84b69530288143e38b643b9',
                 'sample-2.out': '9ae0ea9e3c9c6e1b9b6252c8395efdc1',
             })
+    def test_call_download_atcoder_abc114_c(self):
+        self.snippet_call_download('https://atcoder.jp/contests/abc114/tasks/abc114_c', [
+                {"input": "575\n", "output": "4\n"},
+                {"input": "3600\n", "output": "13\n"},
+                {"input": "999999999\n", "output": "26484\n"},
+            ], type='json')