Skip to content

Commit 908d9a7

Browse files
committed
small improvement in dataexchange unittest
1 parent 078bd3c commit 908d9a7

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

test/core_extend_dataexchange_unittest.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import unittest
2222

2323
from OCC.Core.BRepPrimAPI import BRepPrimAPI_MakeTorus
24+
from OCC.Core.TopoDS import TopoDS_Compound
2425

2526
from OCC.Extend.DataExchange import (read_step_file,
2627
read_step_file_with_names_colors,
@@ -56,8 +57,11 @@ def test_read_step_file(self):
5657

5758

5859
def test_read_step_file_multiple_shape_as_root(self):
59-
read_step_file(STEP_MULTIPLE_ROOT, as_compound=True)
60-
read_step_file(STEP_MULTIPLE_ROOT, as_compound=False)
60+
t = read_step_file(STEP_MULTIPLE_ROOT, as_compound=True)
61+
self.assertTrue(isinstance(t, TopoDS_Compound))
62+
63+
l = read_step_file(STEP_MULTIPLE_ROOT, as_compound=False)
64+
self.assertEqual(len(l), 3)
6165

6266

6367
def test_read_step_file_names_colors(self):

0 commit comments

Comments
 (0)