Add one more synthetic env
This commit is contained in:
@@ -15,6 +15,7 @@ if str(lib_dir) not in sys.path:
|
||||
|
||||
from datasets import QuadraticFunc
|
||||
from datasets import ConstantFunc
|
||||
from datasets import DynamicLinearFunc
|
||||
from datasets import DynamicQuadraticFunc
|
||||
from datasets import ComposedSinFunc
|
||||
|
||||
@@ -50,3 +51,20 @@ class TestDynamicFunc(unittest.TestCase):
|
||||
|
||||
function.set_timestamp(1)
|
||||
print(function(2))
|
||||
|
||||
def test_simple_linear(self):
|
||||
timestamps = 30
|
||||
function = DynamicLinearFunc()
|
||||
function_param = dict()
|
||||
function_param[0] = ComposedSinFunc(
|
||||
num=timestamps, num_sin_phase=4, phase_shift=1.0, max_amplitude=1.0
|
||||
)
|
||||
function_param[1] = ConstantFunc(constant=0.9)
|
||||
function.set(function_param)
|
||||
print(function)
|
||||
|
||||
with self.assertRaises(TypeError) as context:
|
||||
function(0)
|
||||
|
||||
function.set_timestamp(1)
|
||||
print(function(2))
|
||||
|
Reference in New Issue
Block a user