iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update LFNA version 1.0

Browse Source
This commit is contained in:
D-X-Y
2021-05-13 21:33:34 +08:00
parent 3d3a04705f
commit cfabd05de8
11 changed files with 340 additions and 299 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_synthetic_env.py
View File

@@ -23,8 +23,16 @@ class TestSynethicEnv(unittest.TestCase):
def test_simple(self):
mean_generator = ComposedSinFunc(constant=0.1)
std_generator = ConstantFunc(constant=0.5)
dataset = SyntheticDEnv([mean_generator], [[std_generator]], num_per_task=5000)
print(dataset)
for timestamp, tau in dataset:
assert tau.shape == (5000, 1)
self.assertEqual(tau.shape, (5000, 1))
def test_length(self):
mean_generator = ComposedSinFunc(constant=0.1)
std_generator = ConstantFunc(constant=0.5)
dataset = SyntheticDEnv([mean_generator], [[std_generator]], num_per_task=5000)
self.assertEqual(len(dataset), 100)
dataset = SyntheticDEnv([mean_generator], [[std_generator]], mode="train")
self.assertEqual(len(dataset), 60)