feat(poly): Added poly data generation

dist/generate.py

@@ -163,13 +163,14 @@ def generate_test_cases():
 
 def main():
 	parser = argparse.ArgumentParser()
-	parser.add_argument('-o', '--output', type=str, help="Output file")
+	parser.add_argument('-o', '--output', type=str, help='Output file')
 	args = parser.parse_args()
+	output = generate_test_cases()
 	if not args.output:
-		print(generate_test_cases(), end='')
+		print(output, end='')
 	else:
 		with open(args.output, 'w') as file:
-			file.write(generate_test_cases())
+			file.write(output)
 
 
 if __name__ == '__main__':

poly/generate.py

@@ -0,0 +1,107 @@
+#!/usr/bin/env python3
+import argparse
+from concurrent.futures import ProcessPoolExecutor
+from mpmath import mp
+import sympy as sp
+
+mp.dps = 50
+
+precision = 17
+zero_threshold = mp.mpf('1e-40')
+
+nn = 23
+
+def exp(x):
+	return mp.exp(x)
+def sin(x):
+	return mp.sin(x)
+def cos(x):
+	return mp.cos(x)
+def f1(x):
+	return mp.fabs(x) + x/2 - x*x
+def f2(x):
+	return -3*mp.sin(10*x) + 10*mp.sin(mp.fabs(x) + x/2)
+
+def uniform(n):
+	return [0] if n == 1 else [2 * mp.mpf(k) / (n-1) - 1 for k in range(n)]
+def chebyshev(n):
+	return [-mp.cos((2*k - 1) * mp.pi / (2*n)) for k in range(1, n + 1)]
+def chebyshev_2(n):
+	return [mp.sin(mp.pi / 2 * x) for x in uniform(n)]
+
+test_cases = []
+
+functions = [exp, sin, cos, f1, f2]
+distributions = [uniform, chebyshev, chebyshev_2]
+point_counts = [7, 15, 25]
+intervals = [(-2, 2), (-10, 10)]
+
+for func in functions:
+	for dist in distributions:
+		for n in point_counts:
+			for a, b in intervals:
+				test_cases.append((func, dist, n, a, b))
+
+
+def stretched(points, a, b):
+	if not points:
+		return []
+	if len(points) == 1 or min(points) == max(points):
+		return [(a+b)/2] * len(points)
+	return [a + (p-min(points)) * (b-a) / (max(points)-min(points)) for p in points]
+
+
+def format_number(number):
+	return '0' if abs(number) < zero_threshold else mp.nstr(number, n=precision).removesuffix('.0')
+
+
+def format_array(array):
+	return '[' + ', '.join(format_number(x) for x in array) + ']'
+
+
+def format_test_case(func, dist, X, Y, coeffs, xx, yy):
+	return '\n'.join([
+		'[[test_cases]]',
+		f'func = "{func.__name__}"',
+		f'dist = "{dist.__name__}"',
+		f'X = {format_array(X)}',
+		f'Y = {format_array(Y)}',
+		f'coeffs = {format_array(coeffs)}',
+		f'xx = {format_array(xx)}',
+		f'yy = {format_array(yy)}',
+	])
+
+
+def generate_test_case(params):
+	func, dist, n, a, b = params
+	X = stretched(dist(n), a, b)
+	Y = [func(x) for x in X]
+	x_sym = sp.symbols('x')
+	polynomial = sp.interpolate(list(zip(X, Y)), x_sym)
+	coeffs = [polynomial.coeff(x_sym, i) for i in reversed(range(n))]
+	f = sp.lambdify(x_sym, polynomial, modules='mpmath')
+	xx = stretched(uniform(nn), a, b)
+	yy = [f(x) for x in xx]
+	return format_test_case(func, dist, X, Y, coeffs, xx, yy)
+
+
+def generate_test_cases():
+	with ProcessPoolExecutor() as executor:
+		results = list(executor.map(generate_test_case, test_cases))
+	return '\n\n'.join(results)
+
+
+def main():
+	parser = argparse.ArgumentParser()
+	parser.add_argument('-o', '--output', type=str, help='Output file')
+	args = parser.parse_args()
+	output = generate_test_cases()
+	if not args.output:
+		print(output, end='')
+	else:
+		with open(args.output, 'w') as file:
+			file.write(output)
+
+
+if __name__ == '__main__':
+	main()

requirements.txt

@@ -1 +1,2 @@
-mpmath
+mpmath
+sympy

verify_data.py

@@ -11,4 +11,5 @@ def check_data(original, generator):
 		execute_command([generator, '-o', tmpfile.name])
 		execute_command(['diff', original, tmpfile.name])
 
-check_data('dist/dist.toml', 'dist/generate.py')
+check_data('dist/dist.toml', 'dist/generate.py')
+check_data('poly/poly.toml', 'poly/generate.py')