Test practice-b-*

Author: qryxip

.github/workflows/ci.yml

@@ -57,6 +57,11 @@ jobs:
       - name: Checkout
         uses: actions/checkout@v1
 
+      - name: setup-python
+        uses: actions/setup-python@v1
+        with:
+          python-version: '3.8'
+
       - name: rust-toolchain
         uses: actions-rs/toolchain@v1
         with:

examples/practice-b-naive.rs

@@ -2,7 +2,7 @@
 
 use maplit::hashset;
 
-use std::io;
+use std::{io, str};
 
 fn main() {
     fn read_line() -> String {
@@ -28,7 +28,7 @@ fn main() {
         5 => on_5(query),
         _ => unreachable!(),
     };
-    println!("! {}", String::from_utf8(ans).unwrap());
+    println!("! {}", str::from_utf8(&ans).unwrap());
 }
 
 fn on_26(mut query: impl FnMut(u8, u8) -> bool) -> Vec<u8> {

examples/practice-b-proconio.rs

@@ -4,7 +4,7 @@ use maplit::hashset;
 use proconio::input;
 use proconio::source::line::LineSource;
 
-use std::io;
+use std::{io, str};
 
 fn main() {
     let stdin = io::stdin();
@@ -29,7 +29,7 @@ fn main() {
         5 => on_5(query),
         _ => unreachable!(),
     };
-    println!("! {}", String::from_utf8(ans).unwrap());
+    println!("! {}", str::from_utf8(&ans).unwrap());
 }
 
 fn on_26(mut query: impl FnMut(u8, u8) -> bool) -> Vec<u8> {
@@ -100,6 +100,8 @@ mod tests {
                 let wr = balls.iter().position(|&b| b == r).unwrap();
                 wl < wr
             });
+            let ans = str::from_utf8(&ans).unwrap();
+            let balls = str::from_utf8(&balls).unwrap();
             assert_eq!(ans, balls);
             assert!(queries <= 7);
         }

examples/practice-b-text-io.rs

@@ -3,7 +3,7 @@
 use maplit::hashset;
 use text_io::{read, try_read, try_scan};
 
-use std::io;
+use std::{io, str};
 
 #[allow(clippy::try_err)]
 fn main() {
@@ -28,7 +28,7 @@ fn main() {
         5 => on_5(query),
         _ => unreachable!(),
     };
-    println!("! {}", String::from_utf8(ans).unwrap());
+    println!("! {}", str::from_utf8(&ans).unwrap());
 }
 
 fn on_26(mut query: impl FnMut(u8, u8) -> bool) -> Vec<u8> {
@@ -99,6 +99,8 @@ mod tests {
                 let wr = balls.iter().position(|&b| b == r).unwrap();
                 wl < wr
             });
+            let ans = str::from_utf8(&ans).unwrap();
+            let balls = str::from_utf8(&balls).unwrap();
             assert_eq!(ans, balls);
             assert!(queries <= 7);
         }

examples/practice-b-whiteread.rs

@@ -3,6 +3,8 @@
 use maplit::hashset;
 use whiteread::Reader;
 
+use std::str;
+
 fn main() {
     let mut rdr = Reader::from_stdin_naive();
 
@@ -18,7 +20,7 @@ fn main() {
         5 => on_5(query),
         _ => unreachable!(),
     };
-    println!("! {}", String::from_utf8(ans).unwrap());
+    println!("! {}", str::from_utf8(&ans).unwrap());
 }
 
 fn on_26(mut query: impl FnMut(u8, u8) -> bool) -> Vec<u8> {
@@ -89,6 +91,8 @@ mod tests {
                 let wr = balls.iter().position(|&b| b == r).unwrap();
                 wl < wr
             });
+            let ans = str::from_utf8(&ans).unwrap();
+            let balls = str::from_utf8(&balls).unwrap();
             assert_eq!(ans, balls);
             assert!(queries <= 7);
         }

examples/testers/practice-b.py

@@ -0,0 +1,58 @@
+import itertools
+import random
+import string
+from argparse import ArgumentParser
+from subprocess import Popen, PIPE
+from typing import List
+
+
+def main() -> None:
+    parser = ArgumentParser()
+    parser.add_argument('bin')
+
+    binary = parser.parse_args().bin
+
+    for _ in range(100):
+        judge(binary, ''.join(random.sample(string.ascii_uppercase, 26)))
+
+    for balls in itertools.permutations(string.ascii_uppercase[:5]):
+        judge(binary, ''.join(balls))
+
+
+def judge(binary: str, balls: str) -> None:
+    n = len(balls)
+    q = 7 if n == 5 else 100
+
+    with Popen([binary], stdin=PIPE, stdout=PIPE) as proc:
+        def read_words() -> List[str]:
+            return proc.stdout.readline().decode('utf-8').split()
+
+        def on_query(c1: str, c2: str) -> None:
+            reply = '<' if balls.index(c1) < balls.index(c2) else '>'
+            proc.stdin.write(f'{reply}\n'.encode('utf-8'))
+            proc.stdin.flush()
+
+        def on_answer(ans: str) -> None:
+            if ans != balls:
+                raise Exception('wrong answer')
+
+        proc.stdin.write(f'{n} {q}\n'.encode('utf-8'))
+        proc.stdin.flush()
+
+        for _ in range(q):
+            words = read_words()
+            if len(words) == 3 and words[0] == '?':
+                on_query(words[1], words[2])
+            elif len(words) == 2 and words[0] == '!':
+                return on_answer(words[1])
+            else:
+                raise Exception('invalid')
+        else:
+            words = read_words()
+            if len(words) == 2 and words[0] == '!':
+                return on_answer(words[1])
+            raise Exception('answer me')
+
+
+if __name__ == '__main__':
+    main()