백준 1330번 두 수 비교하기

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

Author: skysign

백준 1330번 두 수 비교하기/main.py

@@ -0,0 +1,13 @@
+def solve():
+    n1, n2 = input().split(' ')
+    n1, n2 = int(n1), int(n2)
+    if n1 < n2:
+        print('<')
+    elif n1 > n2:
+        print('>')
+    else:
+        print('==')
+
+
+if __name__ == '__main__':
+    solve()

백준 1330번 두 수 비교하기/test1.txt

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

백준 1330번 두 수 비교하기/test_main.py

@@ -0,0 +1,15 @@
+import sys
+from pathlib import Path
+from unittest import TestCase
+from main import solve
+
+
+class Test(TestCase):
+    def test1_solve(self):
+        sys.stdin = open('test1.txt', 'r')
+        stdout = sys.stdout
+        sys.stdout = open('stdout.txt', 'w')
+        solve()
+        sys.stdout.close()
+        sys.stdout = stdout
+        self.assertEqual('<', Path('stdout.txt').read_text().strip())