diff --git a/test/function_test.py b/test/function_test.py new file mode 100644 index 0000000..ba38383 --- /dev/null +++ b/test/function_test.py @@ -0,0 +1,48 @@ +# -*- coding: utf-8 -*- +""" +>>> import random +>>> import math +>>> from samila import GenerativeImage, Projection +>>> import pickle +>>> def f1(x,y): +... result = random.uniform(-1,1) * x**2 - math.sin(y**2) + abs(y-x) +... return result +>>> def f2(x,y): +... result = random.uniform(-1,1) * y**3 - math.cos(x**2) + 2*x +... return result +>>> g = GenerativeImage(f1,f2) +>>> g.function1 == f1 +True +>>> g.function2 == f2 +True +>>> g.fig +>>> g.generate() +>>> isinstance(g.data1, list) +True +>>> isinstance(g.data2, list) +True +>>> g.generate(seed=10, start=-2*math.pi, step=0.1, stop=math.pi/2) +>>> g.seed +10 +>>> with open("test/test1_1_d1.pkl", "rb") as fp: +... temp_data = pickle.load(fp) +>>> temp_data == g.data1 +True +>>> with open("test/test1_1_d2.pkl", "rb") as fp: +... temp_data = pickle.load(fp) +>>> temp_data == g.data2 +True +>>> g.plot() +>>> g.plot(color='red') +>>> g.plot(color='red', bgcolor='black') +>>> from samila import GenerativeImage, Projection +>>> g.plot(projection=Projection.POLAR, color='red', bgcolor='black') +>>> g.plot(projection=Projection.POLAR, color=(.1, .2, .8)) +>>> g.plot(bgcolor=(.1, .2, .8), spot_size=0.1) +>>> g.plot(size=(20, 20)) +>>> result = g.nft_storage(api_key="") +>>> result['status'] +False +>>> result['message'] +'API Key is missing, make sure the `Authorization` header has a value in the following format `Bearer {api key}`.' +""" diff --git a/test/test1_1_d1.pkl b/test/test1_1_d1.pkl new file mode 100644 index 0000000..214d566 Binary files /dev/null and b/test/test1_1_d1.pkl differ diff --git a/test/test1_1_d2.pkl b/test/test1_1_d2.pkl new file mode 100644 index 0000000..b0df2af Binary files /dev/null and b/test/test1_1_d2.pkl differ