def test_to_dict(self): points = [0., 0., 1., 2., 0.5, 2.7, 1.3, 4.0] expected = {} expected["typeid"] = "scanpointgenerator:generator/ArrayGenerator:1.0" expected["axis"] = "x" expected["units"] = "cm" expected["points"] = points expected["alternate"] = True g = ArrayGenerator("x", "cm", points, True) self.assertEqual(expected, g.to_dict())
class JArrayGenerator(JavaIteratorWrapper): """ Create an ArrayGenerator and wrap the points into java Scalar objects """ def __init__(self, name, units, points): super(JArrayGenerator, self).__init__() points = points.tolist() # Convert from array to list self.name = name self.generator = ArrayGenerator(name, units, points) logging.debug(self.generator.to_dict()) def _iterator(self): for point in self.generator.iterator(): name = self.name[0] index = point.indexes[0] position = point.positions[name] java_point = Scalar(name, index, position) yield java_point