백준 15652번 N과 M (4)

문제 링크: https://www.acmicpc.net/problem/15652

Author: skysign

백준 15652번 N과 M (4)/test1.txt

@@ -0,0 +1 @@
+3 1

백준 15652번 N과 M (4)/test1_answer.txt

@@ -0,0 +1,3 @@
+1
+2
+3

백준 15652번 N과 M (4)/test2.txt

@@ -0,0 +1 @@
+4 2

백준 15652번 N과 M (4)/test2_answer.txt

@@ -0,0 +1,10 @@
+1 1
+1 2
+1 3
+1 4
+2 2
+2 3
+2 4
+3 3
+3 4
+4 4

백준 15652번 N과 M (4)/test3.txt

@@ -0,0 +1 @@
+3 3

백준 15652번 N과 M (4)/test3_answer.txt

@@ -0,0 +1,10 @@
+1 1 1
+1 1 2
+1 1 3
+1 2 2
+1 2 3
+1 3 3
+2 2 2
+2 2 3
+2 3 3
+3 3 3

백준 15652번 N과 M (4)/test_main.py

@@ -0,0 +1,31 @@
+import sys
+from pathlib import Path
+from unittest import TestCase
+from main import solve
+
+def my_solve(testcase_input):
+    sys.stdin = open(testcase_input, 'r')
+    stdout = sys.stdout
+    sys.stdout = open('stdout.txt', 'w')
+    solve()
+    sys.stdout.close()
+    sys.stdout = stdout
+
+class Test(TestCase):
+    def test1_solve(self):
+        my_solve('test1.txt')
+        self.assertEqual(
+            Path('test1_answer.txt').read_text().strip(),
+            Path('stdout.txt').read_text().strip())
+
+    def test2_solve(self):
+        my_solve('test2.txt')
+        self.assertEqual(
+            Path('test2_answer.txt').read_text().strip(),
+            Path('stdout.txt').read_text().strip())
+
+    def test3_solve(self):
+        my_solve('test3.txt')
+        self.assertEqual(
+            Path('test3_answer.txt').read_text().strip(),
+            Path('stdout.txt').read_text().strip())