def setUp(self):
e = NullEngine()
e.start()
e.new_scene()
scene = e.scenes[-1]
s = ParametricSurface()
e.add_source(s)
o = Outline()
s.add_child(o)
o1 = Outline()
s.add_child(o1)
self.scene = scene
self.e = e
self.s = s
self.o = o
self.o1 = o1
return
def setUp(self):
e = NullEngine()
e.start()
e.new_scene()
scene = e.scenes[-1]
s = ParametricSurface()
e.add_source(s)
o = Outline()
s.add_child(o)
o1 = Outline()
s.add_child(o1)
self.scene = scene
self.e = e
self.s = s
self.o = o
self.o1 = o1
return
def test_script_recording(self):
"Does script recording work correctly."
# Create a mayavi pipeline and record it.
tape = self.tape
e = NullEngine()
e.start()
# Start recording.
tape.recording = True
tape.register(e, known=True, script_id="engine")
e.new_scene()
# print tape.script
self.assertEqual(tape.lines[-1], "dummy_viewer = engine.new_scene()")
src = ParametricSurface()
e.add_source(src)
expect = "from enthought.mayavi.sources.parametric_surface " "import ParametricSurface"
self.assertEqual(tape.lines[-3], expect)
self.assertEqual(tape.lines[-2], "parametric_surface = ParametricSurface()")
self.assertEqual(tape.lines[-1], "engine.add_source(parametric_surface)")
src.function = "dini"
self.assertEqual(tape.lines[-1], "parametric_surface.function = 'dini'")
o = Outline()
e.add_module(o)
expect = "from enthought.mayavi.modules.outline import Outline"
self.assertEqual(tape.lines[-3], expect)
self.assertEqual(tape.lines[-2], "outline = Outline()")
self.assertEqual(tape.lines[-1], "engine.add_module(outline)")
o.actor.property.color = (1, 0, 0)
self.assertEqual(tape.lines[-1], "outline.actor.property.color = (1.0, 0.0, 0.0)")
s = Surface()
e.add_module(s)
expect = "from enthought.mayavi.modules.surface import Surface"
self.assertEqual(tape.lines[-3], expect)
self.assertEqual(tape.lines[-2], "surface = Surface()")
self.assertEqual(tape.lines[-1], "engine.add_module(surface)")
s.actor.property.representation = "wireframe"
self.assertEqual(tape.lines[-1], "surface.actor.property.representation = 'wireframe'")
o.actor.property.representation = "wireframe"
self.assertEqual(tape.lines[-1], "outline.actor.property.representation = 'wireframe'")
s.actor.property.opacity = 0.5
self.assertEqual(tape.lines[-1], "surface.actor.property.opacity = 0.5")
s.actor.mapper.scalar_visibility = False
self.assertEqual(tape.lines[-1], "surface.actor.mapper.scalar_visibility = False")
# print tape.script
# Stop recording and test.
tape.unregister(e)
tape.record("#end") # Placeholder
o.actor.property.opacity = 0.5
self.assertEqual(tape.lines[-1], "#end")
s.actor.property.color = (1, 0, 0)
self.assertEqual(tape.lines[-1], "#end")
s.enable_contours = True
self.assertEqual(tape.lines[-1], "#end")
src.function = "klein"
self.assertEqual(tape.lines[-1], "#end")
