Skip to content

Commit 44df852

Browse files
authored
Merge pull request #842 from Kaggle/rm-dipy-deprecated-method
Remove use of deprecated Renderer.
2 parents d87710b + b641e32 commit 44df852

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

tests/test_dipy.py

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -8,22 +8,20 @@
88
from dipy.denoise.enhancement_kernel import EnhancementKernel
99
from dipy.tracking.fbcmeasures import FBCMeasures
1010
from dipy.core.sphere import Sphere
11-
from dipy.viz import window
12-
from xvfbwrapper import Xvfb
11+
from dipy.viz import window, actor
1312

1413
class TestDipy(unittest.TestCase):
15-
def test_renderer(self):
16-
vdisplay = Xvfb()
17-
vdisplay.start()
14+
def test_scene(self):
15+
xyz = 10 * np.random.rand(100, 3)
16+
colors = np.random.rand(100, 4)
17+
radii = np.random.rand(100) + 0.5
1818

19-
ren = window.Renderer()
19+
sphere_actor = actor.sphere(centers=xyz, colors=colors, radii=radii)
2020

21-
with tempfile.TemporaryDirectory() as dir:
22-
out_file = os.path.join(dir, 'test.png')
23-
window.record(ren, n_frames=1, out_path=out_file, size=(600, 600))
24-
self.assertTrue(os.path.exists(out_file))
25-
26-
vdisplay.stop()
21+
scene = window.Scene()
22+
scene.add(sphere_actor)
23+
24+
self.assertEqual((0, 0), scene.GetSize())
2725

2826
def test_graph(self):
2927
g = Graph()

0 commit comments

Comments
 (0)