def test_to_dict(self):
g = StaticPointGenerator(7)
expected_dict = {
"typeid": "scanpointgenerator:generator/StaticPointGenerator:1.0",
"size": 7,
}
self.assertEqual(expected_dict, g.to_dict())
def test_to_dict_with_axis(self):
g = StaticPointGenerator(7, 'repeats')
expected_dict = {
"typeid": "scanpointgenerator:generator/StaticPointGenerator:1.0",
"size": 7,
"axes": ['repeats'],
}
self.assertEqual(expected_dict, g.to_dict())