def test_has_children():
with CATIADocHandler(cat_product) as handler:
product = handler.document.product()
assert product.has_children()
with CATIADocHandler(cat_part_measurable) as handler:
part_product = handler.document.product()
assert not part_product.has_children()
def test_is_updated():
with CATIADocHandler(cat_part_measurable) as handler:
part = handler.document.part()
assert part.is_upated(part.part)
with CATIADocHandler(cat_part_not_updated) as handler:
part = handler.document.part()
assert not part.is_upated(part.part)
def test_is_catproduct_is_catpart():
with CATIADocHandler(cat_product) as handler:
product = handler.document.product()
assert product.is_catproduct()
assert not product.is_catpart()
with CATIADocHandler(cat_part_measurable) as handler:
part = handler.document.product()
assert part.is_catpart()
assert not part.is_catproduct()
def test_paper_size():
with CATIADocHandler(cat_drawing) as handler:
catia = handler.catia
drawing = DrawingRoot(catia)
sheet_1 = drawing.sheets[0]
assert sheet_1.paper_size == 2
sheet_1.paper_size = 5
assert sheet_1.paper_size == 5
with pytest.raises(CATIAApplicationException):
with CATIADocHandler(cat_drawing) as handler:
catia = handler.catia
drawing = DrawingRoot(catia)
sheet_1 = drawing.sheets[0]
sheet_1.paper_size = 10
def test_count_types():
with CATIADocHandler(cat_product) as handler:
documents = handler.documents
num = documents.count_types('.catpart')
assert num == 2
def test_add_documents():
with CATIADocHandler(new_document='Part') as handler:
document = handler.document
assert 'CATPart' in document.name
with CATIADocHandler(new_document='Part') as handler:
document = handler.document
assert 'CATPart' in document.name
with CATIADocHandler(new_document='Part') as handler:
document = handler.document
assert 'CATPart' in document.name
with pytest.raises(ValueError):
with CATIADocHandler(new_document='lala'):
pass
def test_product():
with CATIADocHandler(cat_product) as handler:
document = handler.document
product = document.product()
assert product.name in cat_product.name
assert document.is_product
assert not document.is_part
def test_part():
with CATIADocHandler(cat_part_measurable) as handler:
document = handler.document
part = document.part()
assert part.name == "CF_catia_measurable_part"
assert document.is_part
assert not document.is_product
def test_path():
with CATIADocHandler(cat_part_measurable) as handler:
part_product = handler.document.product()
product_path = Path(os.getcwd(), cat_part_measurable)
assert product_path == part_product.path()
def test_relations_count():
with CATIADocHandler(cat_part_3) as handler:
document = handler.document
part = document.part()
relations = Relations(part.relations_com_obj())
assert relations.count == 6
def test_get_products():
with CATIADocHandler(cat_product) as handler:
product = handler.document.product()
products = product.get_products()
assert 'Product(part_number="CF_SubProduct1", file_name="CF_SubProduct1.CATProduct")' == products[
0].__repr__()
def test_get_points_on_curve():
with CATIADocHandler(cat_part_measurable) as handler:
catia = handler.catia
document = handler.document
part = document.part()
spa_workbench = create_spa_workbench(document.document)
hybrid_body = part.get_hybrid_body_by_name('Lines')
line1 = hybrid_body.HybridShapes.Item(1)
line1_reference = part.create_reference(line1)
line1_measurable = create_measurable(spa_workbench, line1_reference)
catia_measurable_line1 = CATIAMeasurable(line1_measurable)
points_on_curve = (
0.0,
8.0,
-4,
43.0,
17.5,
-16.5,
86.0,
27,
-29,
)
catia_points_on_curve = catia_measurable_line1.get_points_on_curve(
catia)
catia_points_on_curve = round_tuple(catia_points_on_curve, 6)
assert points_on_curve == catia_points_on_curve
def test_find_object_by_name():
with CATIADocHandler(cat_part_measurable) as handler:
part = handler.document.part()
item = part.find_object_by_name('Extrude.1')
assert item.name == 'Extrude.1'
def test_count_parameters():
with CATIADocHandler(cat_part_blank) as handler:
document = handler.document
part = document.part()
parameters = Parameters(part.parameters_com_obj())
assert parameters.count_parameters() == 5
def test_get_points_on_axis():
with CATIADocHandler(cat_part_measurable) as handler:
catia = handler.catia
document = handler.document
part = document.part()
spa_workbench = create_spa_workbench(document.document)
hybrid_body = part.get_hybrid_body_by_name('Cylinders')
cylinder = hybrid_body.HybridShapes.Item(1)
cylinder_reference = part.create_reference(cylinder)
cylinder_measurable = create_measurable(spa_workbench,
cylinder_reference)
catia_measurable_cylinder = CATIAMeasurable(cylinder_measurable)
cylinder = (
-92.049,
142.675,
10.000,
-92.049,
142.675,
20.000,
-92.049,
142.675,
0.000,
)
catia_cylinder = catia_measurable_cylinder.get_points_on_axis(catia)
catia_cylinder = round_tuple(catia_cylinder, 6)
assert cylinder == catia_cylinder
def test_center():
"""
:return:
"""
with CATIADocHandler(cat_part_measurable) as handler:
catia = handler.catia
document = handler.document
center = (-47.039, 83.488, 0.0)
spa_workbench = create_spa_workbench(document.document)
part = document.part()
hybrid_body = part.get_hybrid_body_by_name('Arcs')
arc = hybrid_body.HybridShapes.Item(1)
arc_reference = part.create_reference(arc)
arc_measurable = create_measurable(spa_workbench, arc_reference)
catia_measurable_arc = CATIAMeasurable(arc_measurable)
catia_center = catia_measurable_arc.get_center(catia)
assert center == catia_center
def test_relations_create_set_of_relations():
with CATIADocHandler(cat_part_blank) as handler:
document = handler.document
part = document.part()
relations = Relations(part.relations_com_obj())
relations.create_set_of_relations(relations)
assert relations.name == 'Relations'
def test_create_geometrical_set():
with CATIADocHandler(cat_part_measurable) as handler:
document = handler.document
part = document.part()
geometrical_set = part.create_geometrical_set('lala')
assert geometrical_set.name == 'lala'
def test_relations_item():
with CATIADocHandler(cat_part_3) as handler:
document = handler.document
part = document.part()
relations = Relations(part.relations_com_obj())
relation = relations.item(0)
assert relation.name == 'Formula.1'
def test_annotation_sets():
with CATIADocHandler(cat_part_measurable) as handler:
document = handler.document
part = document.part()
annotation_sets = part.get_annotation_sets()
assert annotation_sets.Item(1).name == 'Annotation Set.1'
def test_parameters_name():
with CATIADocHandler(cat_part_3) as handler:
document = handler.document
part = document.part()
parameters = Parameters(part.parameters_com_obj())
first_parameter = parameters.item(0)
assert first_parameter.name == r'CF_Part3\PartBody\Pad.1\FirstLimit\Length'
def test_in_work_object():
with CATIADocHandler(cat_part_measurable) as handler:
part = handler.document.part()
planes_hybrid_body = part.get_hybrid_body_by_name('Planes')
part.in_work_object = planes_hybrid_body
assert part.in_work_object.name == 'Planes'
def test_relations_create_rule_base():
with CATIADocHandler(cat_part_blank) as handler:
document = handler.document
part = document.part()
relations = Relations(part.relations_com_obj())
rule_base = relations.create_rule_base('new-rule-base')
assert rule_base.name == 'new-rule-base'
def test_get_bodies_names():
with CATIADocHandler(cat_part_measurable) as handler:
document = handler.document
part = document.part()
body_names = part.get_bodies_names()
assert body_names[0] == 'PartBody'
assert body_names[1] == 'AnotherPartBody'
def test_view_scale():
with CATIADocHandler(cat_drawing) as handler:
sheets = DrawingRoot(handler.catia).sheets
sheet_1 = sheets[0]
views = sheet_1.views
view = views.get_view_by_name('Isometric view')
assert view.scale == 1.0
view.scale = 2.0
assert view.scale == 2.0
def test_create_list():
with CATIADocHandler(cat_part_blank) as handler:
document = handler.document
part = document.part()
parameters = Parameters(part.parameters_com_obj())
new_list = parameters.create_list('new_list')
assert new_list.name == r'new_list'
def test_orientation():
with CATIADocHandler(cat_drawing) as handler:
catia = handler.catia
drawing = DrawingRoot(catia)
sheets = drawing.sheets
sheet_1 = sheets[0]
assert sheet_1.orientation == 'catPaperLandscape'
sheet_1.orientation = 0
assert sheet_1.orientation == 'catPaperPortrait'
def test_scale():
with CATIADocHandler(cat_drawing) as handler:
catia = handler.catia
drawing = DrawingRoot(catia)
sheets = drawing.sheets
sheet_1 = sheets[0]
assert sheet_1.scale == 1.0
sheet_1.scale = 2.0
assert sheet_1.scale == 2.0
def test_get_axes_names():
with CATIADocHandler(cat_part_measurable) as handler:
document = handler.document
part = document.part()
axes_names = part.get_axes_names()
assert axes_names[0] == 'Axis System.3'
assert axes_names[1] == 'Axis System.4'
def test_relations_get_item_by_index():
with CATIADocHandler(cat_part_3) as handler:
document = handler.document
part = document.part()
relations = Relations(part.relations_com_obj())
item = relations.get_item_by_index(0)
assert item.name == 'Formula.1'
