Merge branch 'master' of https://github.com/luogu-dev/cyaron

Author: kkksc03

README.md

@@ -2,7 +2,7 @@
 CYaRon: Yet Another Random Olympic-iNformatics test data generator
 
 By Luogu
-![](https://travis-ci.org/luogu-dev/cyaron.svg?branch=master)
+[![](https://travis-ci.org/luogu-dev/cyaron.svg?branch=master)](https://travis-ci.org/luogu-dev/cyaron)
 
 你是否遇到以下情况：
 - 希望在5分钟内写出一组随机数据

cyaron/io.py

@@ -57,11 +57,26 @@ def __del__(self):
         except Exception:
             pass
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """__del__(self) -> None
+            Exit the context of the IO object and close the input file and the output file
+        """
+        try:
+            self.input_file.close()
+            self.output_file.close()
+        except Exception:
+            pass
+
     @staticmethod
     def __write(file, *args, **kwargs):
-        """__write(file, *args) -> None
+        """__write(file, *args, **kwargs) -> None
             Write every element in *args into file. If the element isn't "\n", insert a space. It will convert every element into str
             file file -> the file object to write
+            **kwargs:
+                str separator = " " -> a string used to separate every element
         """
         separator = kwargs.get("separator", " ")
         for arg in args:
@@ -73,14 +88,18 @@ def __write(file, *args, **kwargs):
                     file.write(separator)
 
     def input_write(self, *args, **kwargs):
-        """input_write(self, *args) -> None
+        """input_write(self, *args, **kwargs) -> None
             Write every element in *args into the input file. Splits with spaces. It will convert every element into string
+            **kwargs:
+                str separator = " " -> a string used to separate every element
         """
         IO.__write(self.input_file, *args, **kwargs)
 
     def input_writeln(self, *args, **kwargs):
-        """input_writeln(self, *args) -> None
+        """input_writeln(self, *args, **kwargs) -> None
             Write every element in *args into the input file and turn to a new line. Splits with spaces. It will convert every element into string
+            **kwargs:
+                str separator = " " -> a string used to separate every element
         """
         args = list(args)
         args.append("\n")
@@ -93,19 +112,23 @@ def output_gen(self, shell_cmd):
         """
         self.input_file.close()
         with open(self.input_filename, 'r') as f:
-            self.output_file.write(subprocess.check_output(shell_cmd, shell=True, stdin=f))
+            self.output_file.write(subprocess.check_output(shell_cmd, shell=True, stdin=f).decode('ascii'))
 
         self.input_file = open(self.input_filename, 'a')
 
     def output_write(self, *args, **kwargs):
-        """output_write(self, *args) -> None
+        """output_write(self, *args, **kwargs) -> None
             Write every element in *args into the output file. Splits with spaces. It will convert every element into string
+            **kwargs:
+                str separator = " " -> a string used to separate every element
         """
         IO.__write(self.output_file, *args, **kwargs)
 
     def output_writeln(self, *args, **kwargs):
-        """output_writeln(self, *args) -> None
+        """output_writeln(self, *args, **kwargs) -> None
             Write every element in *args into the output file and turn to a new line. Splits with spaces. It will convert every element into string
+            **kwargs:
+                str separator = " " -> a string used to separate every element
         """
         args = list(args)
         args.append("\n")

cyaron/sequence.py

@@ -4,8 +4,8 @@ class Sequence:
     """Class Sequence: the tool class for sequences.
     """
 
-    def __init__(self, formula, initial_values=[]):
-        """__init__(self, formula, initial_values=[]) -> None
+    def __init__(self, formula, initial_values=()):
+        """__init__(self, formula, initial_values=() -> None
             Create a sequence object.
             int formula(int, function) -> the formula function ...
         """

cyaron/tests/__init__.py

@@ -1 +1,3 @@
-from .sequence_test import TestSequence
+from .sequence_test import TestSequence
+from .io_test import TestIO
+from .str_test import TestString

cyaron/tests/io_test.py

@@ -0,0 +1,51 @@
+import unittest
+import os
+import shutil
+import tempfile
+from cyaron import IO
+
+
+class TestIO(unittest.TestCase):
+
+    def setUp(self):
+        self.temp_directory = tempfile.mkdtemp()
+        os.chdir(self.temp_directory)
+
+    def tearDown(self):
+        shutil.rmtree(self.temp_directory)
+
+    def test_create_files_simple(self):
+        IO("test_simple.in", "test_simple.out")
+        self.assertTrue(os.path.exists("test_simple.in"))
+        self.assertTrue(os.path.exists("test_simple.out"))
+
+    def test_create_files_prefix_id(self):
+        IO(file_prefix="test_prefix", data_id=233, input_suffix=".inp", output_suffix=".ans")
+        self.assertTrue(os.path.exists("test_prefix233.inp"))
+        self.assertTrue(os.path.exists("test_prefix233.ans"))
+
+    def test_write_stuff(self):
+        with IO("test_write.in", "test_write.out") as test:
+            test.input_write(1, 2, 3)
+            test.input_writeln([4, 5, 6])
+            test.input_writeln(7, [8, 9])
+            test.output_write([9, 8], 7)
+            test.output_writeln(6, 5, 4)
+            test.output_writeln([3], 2, [1])
+
+        with open("test_write.in") as f:
+            input = f.read()
+        with open("test_write.out") as f:
+            output = f.read()
+        self.assertEqual(input.split(), ['1', '2', '3', '4', '5', '6', '7', '8', '9'])
+        self.assertEqual(output.split(), ['9', '8', '7', '6', '5', '4', '3', '2', '1'])
+        self.assertEqual(input.count("\n"), 2)
+        self.assertEqual(output.count("\n"), 2)
+
+    def test_output_gen(self):
+        with IO("test_gen.in", "test_gen.out") as test:
+            test.output_gen("echo 233")
+
+        with open("test_gen.out") as f:
+            output = f.read()
+        self.assertEqual(output.strip("\n"), "233")

cyaron/tests/sequence_test.py

@@ -1,5 +1,4 @@
 import unittest
-
 from cyaron import Sequence
 
 

cyaron/tests/str_test.py

@@ -0,0 +1,20 @@
+import unittest
+from cyaron import String
+
+
+class TestString(unittest.TestCase):
+
+    def test_random_word(self):
+        test = String.random(100, charset="abc")
+        self.assertTrue(all(a in "abc" for a in test))
+
+    def test_random_word_from_dict(self):
+        my_dict = ["lorem", "ipsum", "dolor", "sit", "amet"]
+        test = String.random(None, charset=my_dict)
+        self.assertTrue(test in my_dict)
+
+    def test_random_sentence(self):
+        sentence = String.random_sentence(10, sentence_terminators=".")
+        self.assertTrue(sentence[0].isupper())
+        self.assertTrue(sentence[-1] == ".")
+        self.assertTrue(sentence.count(" ") == 9)

requirements.txt

@@ -2,7 +2,7 @@
 
 setup(
     name='cyaron',
-    version='0.1.5',
+    version='0.1.6',
     keywords='olympic informatics luogu aqours cyaron lovelive sunshine online judge',
     description='CYaRon: Yet Another Random Olympic-iNformatics test data generator, A library for automatically generating test data for Online Judge, Olympic Informatics or automatic application testing',
     license='LGPLv3',