Sequence library

Author: lin-toto

cyaron/__init__.py

@@ -1,6 +1,7 @@
 from .io import IO
 from .graph import Graph, Edge
 from .str import String
+from .sequence import Sequence
 from .utils import *
 from .consts import *
 from random import randint, randrange, uniform, choice, random

cyaron/graph.py

@@ -166,7 +166,7 @@ def hack_spfa(point_count, **kwargs):
         graph = Graph(point_count, directed)
         if point_count % 2 == 1:
             point_to_skip = point_count / 2 + 1
-        half = point_count / 2
+        half = int(point_count / 2)
 
         for i in range(1, half):
             (x, y) = (i, i + 1)

cyaron/io.py

@@ -36,9 +36,12 @@ def __del__(self):
     @staticmethod
     def __write(file, *args):
         for arg in args:
-            file.write(str(arg))
-            if arg != "\n":
-                file.write(" ")
+            if list_like(arg):
+                IO.__write(file, *arg)
+            else:
+                file.write(str(arg))
+                if arg != "\n":
+                    file.write(" ")
 
     def input_write(self, *args):
         IO.__write(self.input_file, *args)

cyaron/sequence.py

@@ -0,0 +1,25 @@
+from .utils import *
+
+class Sequence:
+
+    def __init__(self, formula, initial_values=[]):
+        self.formula = formula
+        if list_like(initial_values):
+            self.values = {k: v for (k, v) in enumerate(initial_values)}
+        elif isinstance(initial_values, dict):
+            self.values = initial_values
+        else:
+            raise Exception("Initial_values must be either a list/tuple or a dict.")
+
+    def __get_one(self, i):
+        if i in self.values:
+            return self.values[i]
+
+        self.values[i] = self.formula(i, self.__get_one)
+        return self.values[i]
+
+    def get(self, left_range, right_range=None):
+        if right_range is None:
+            return self.__get_one(left_range)
+
+        return [self.__get_one(i) for i in range(left_range, right_range+1)]

cyaron/str.py

@@ -1,5 +1,6 @@
 from .consts import ALPHABET_SMALL, SENTENCE_SEPARATORS, SENTENCE_TERMINATORS
 from .utils import *
+from functools import reduce
 import random
 
 

cyaron/tests/__init__.py

@@ -0,0 +1 @@
+from sequence_test import TestSequence

cyaron/tests/sequence_test.py

@@ -0,0 +1,24 @@
+import unittest
+
+from cyaron import Sequence
+
+
+class TestSequence(unittest.TestCase):
+
+    def test_simple_get_one(self):
+        seq = Sequence(lambda i, f: 2 * i + 3)
+        self.assertEqual(seq.get(1), 5)
+        self.assertEqual(seq.get(5), 13)
+
+    def test_simple_get_many(self):
+        seq = Sequence(lambda i, f: 3 * i + 2)
+        self.assertEqual(seq.get(1, 3), [5, 8, 11])
+
+    def test_func_get_one(self):
+        seq = Sequence(lambda i, f: 2 * i + 3 * f(i - 1), [0])
+        self.assertEqual(seq.get(5), 358)
+
+    def test_func_get_many(self):
+        seq = Sequence(lambda i, f: 3 * i + 2 * f(i - 1), [0])
+        self.assertEqual(seq.get(3, 5), [33, 78, 171])
+

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='cyaron',
-    version='0.1.3',
+    version='0.1.4',
     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',

test_sequence.py

@@ -0,0 +1,9 @@
+#!/usr/bin/env python
+
+from cyaron import *
+
+test = IO("test.in", "test.out")
+
+test.input_writeln(Sequence(lambda i, f: 2*i+1).get(1))
+test.input_writeln(Sequence(lambda i, f: 2*i).get(1, 5))
+test.input_writeln(Sequence(lambda i, f: f(i-1)+1, [0]).get(1, 5))

unit_test.py

@@ -0,0 +1,8 @@
+#!/usr/bin/env python
+
+import unittest
+
+from cyaron.tests import *
+
+if __name__ == '__main__':
+    unittest.main(verbosity=2)