online-judge-tools#438: test rstrip.

fukatani · Oct 1, 2019 · 5763be2 · 5763be2
1 parent 0d2a9cc
commit 5763be2
Showing 1 changed file with 42 additions and 0 deletions.
diff --git a/tests/command_test.py b/tests/command_test.py
@@ -46,6 +46,14 @@ def test_call_test_simple(self):
                     'path': 'test/sample-2.out',
                     'data': 'foo\n'
                 },
+                {
+                    'path': 'test/sample-3.in',
+                    'data': 'foobar \n'
+                },
+                {
+                    'path': 'test/sample-3.out',
+                    'data': 'foobar\n'
+                },
             ],
             expected=[{
                 'status': 'AC',
@@ -65,6 +73,40 @@ def test_call_test_simple(self):
                 },
                 'output': 'bar\n',
                 'exitcode': 0,
+            }, {
+                'status': 'AC',
+                'testcase': {
+                    'name': 'sample-3',
+                    'input': '%s/test/sample-3.in',
+                    'output': '%s/test/sample-3.out',
+                },
+                'output': 'foobar \n',
+                'exitcode': 0,
+            }],
+        )
+
+    def test_call_test_norstrip(self):
+        self.snippet_call_test(
+            args=['-c', cat(), '--no-rstrip'],
+            files=[
+                {
+                    'path': 'test/sample-1.in',
+                    'data': 'foo \n'
+                },
+                {
+                    'path': 'test/sample-1.out',
+                    'data': 'foo\n'
+                },
+            ],
+            expected=[{
+                'status': 'WA',
+                'testcase': {
+                    'name': 'sample-1',
+                    'input': '%s/test/sample-1.in',
+                    'output': '%s/test/sample-1.out',
+                },
+                'output': 'foo \n',
+                'exitcode': 0,
             }],
         )