Add PyPy support

Author: wxh06

.travis.yml

@@ -22,6 +22,8 @@ install:
   - docker pull golang:1  # For go
   - docker pull python:3  # For python3
   - docker pull python:2  # For python2
+  - docker pull pypy:3  # For pypy3
+  - docker pull pypy:2  # For pypy2
   - docker pull node:12  # For node
   - docker pull openjdk  # For javac and java
   - docker pull clangbuiltlinux/ubuntu:llvm10-latest  # For clang-10 & clang++-10

README.md

@@ -31,6 +31,7 @@
   - [`node`](https://nodejs.org/en/download/)
 - [Python](https://www.python.org/)
   - [CPython](https://www.python.org/downloads/)
+  - [PyPy](https://www.pypy.org/)
 
 
 ## Installation
@@ -241,14 +242,22 @@ sudo make install  # python3 setup.py install
     b''
 ]
 >>>
->>> judge(Python(3), b"print('Hello, world!')", [(b'', b'Hello, world!')])  # Python
+>>> judge(Python(3), b"print('Hello, world!')", [(b'', b'Hello, world!')])  # Python 3
 [
     [
         (<Status.AC: 'Accepted'>, (b'Hello, world!\n', b''), 0.05)
     ],
     b"Listing '.'...\n"
     b"Compiling './__init__.py'...\n"
 ]
+>>> judge(PyPy(), b"print('Hello, world!')", [(b'', b'Hello, world!')])  # PyPy 3
+[
+    [
+        (<Status.AC: 'Accepted'>, (b'Hello, world!\n', b''), 0.075)
+    ],
+    b"Listing '.'...\n"
+    b"Compiling './__init__.py'...\n"
+]
 >>>
 >>> judge(Node(12), b'console.log("Hello World")', [(b'', b'Hello World')])  # Node.js
 [

README.zh_Hans_CN.md

@@ -31,6 +31,7 @@
   - [`node`](https://nodejs.org/zh-cn/download/)
 - [Python](https://www.python.org/)
   - [CPython](https://www.python.org/downloads/)
+  - [PyPy](https://www.pypy.org/)
 
 
 ## 安装
@@ -241,14 +242,22 @@ sudo make install  # python3 setup.py install
     b''
 ]
 >>>
->>> judge(Python(3), b"print('Hello, world!')", [(b'', b'Hello, world!')])  # Python
+>>> judge(Python(3), b"print('Hello, world!')", [(b'', b'Hello, world!')])  # Python 3
 [
     [
         (<Status.AC: 'Accepted'>, (b'Hello, world!\n', b''), 0.05)
     ],
     b"Listing '.'...\n"
     b"Compiling './__init__.py'...\n"
 ]
+>>> judge(PyPy(), b"print('Hello, world!')", [(b'', b'Hello, world!')])  # PyPy 3
+[
+    [
+        (<Status.AC: 'Accepted'>, (b'Hello, world!\n', b''), 0.075)
+    ],
+    b"Listing '.'...\n"
+    b"Compiling './__init__.py'...\n"
+]
 >>>
 >>> judge(Node(12), b'console.log("Hello World")', [(b'', b'Hello World')])  # Node.js
 [

dockerjudge/processor.py

@@ -147,6 +147,18 @@ def __init__(self, version=None):
         self.judge = 'java Main'
 
 
+class PyPy(Processor):
+    'PyPy'
+
+    def __init__(self, version=None):
+        pypy = 'pypy' if str(version).startswith('2') else 'pypy3'
+
+        self.image = self._get_image_with_tag('pypy', version)
+        self.source = '__init__.py'
+        self.compile = [pypy, '-m', 'compileall', '.']
+        self.judge = f'{pypy} {self.source}'
+
+
 class Python(Processor):
     'CPython'
 

test_dockerjudge.py

@@ -5,7 +5,8 @@
 import unittest
 
 from dockerjudge import judge
-from dockerjudge.processor import Clang, GCC, Go, Node, OpenJDK, Python, Bash
+from dockerjudge.processor import (Bash, Clang, GCC, Go, Node, OpenJDK,
+                                   PyPy, Python)
 from dockerjudge.status import Status
 
 
@@ -31,6 +32,10 @@ def test_Go(self):
         self.assertEqual(Go(None, {'src': 'golang.go'}).source, 'golang.go')
         self.assertEqual(Go(None, {'bin': 'golang'}).judge, './golang')
 
+    def test_PyPy(self):
+        self.assertEqual(PyPy().compile, PyPy(3).compile)
+        self.assertEqual(PyPy().judge, PyPy(3).judge)
+
 
 class TestDockerJudge(unittest.TestCase):
 
@@ -275,6 +280,28 @@ def test_CE(self):
         self.assertFalse(result[0][1][1][0])
         self.assertFalse(result[0][1][1][1])
 
+    def test_pypy(self):
+        result = judge(
+            PyPy(2),
+            b'a, b = [int(i) for i in raw_input().split()]; print a / b',
+            [(b'1 1', b'1'), (b'1 2', b'0.5'), (b'0 0', b'')]
+        )
+        self.assertEqual(result[0][0][0], Status.AC)
+        self.assertEqual(result[0][1][0], Status.WA)
+        self.assertEqual(result[0][2][0], Status.RE)
+        self.assertTrue(result[0][2][1][1])
+
+    def test_pypy3(self):
+        result = judge(
+            PyPy(3),
+            b'a, b = [int(i) for i in input().split()]; print(a / b)',
+            [(b'1 1', b'1'), (b'1 2', b'0.5'), (b'0 0', b'')]
+        )
+        self.assertEqual(result[0][0][0], Status.WA)
+        self.assertEqual(result[0][1][0], Status.AC)
+        self.assertEqual(result[0][2][0], Status.RE)
+        self.assertTrue(result[0][2][1][1])
+
 
 class TestGoLang(unittest.TestCase):