mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-06 12:26:26 +00:00
67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
|
import json
|
||
|
|
from behave import step
|
||
|
|
from utils import IfcFile, assert_attribute, assert_type
|
||
|
|
|
||
|
|
|
||
|
|
def get_classification(name):
|
||
|
|
classifications = [c for c in IfcFile.get().by_type('IfcClassification') if c.Name == name]
|
||
|
|
if len(classifications) != 1:
|
||
|
|
assert False, f'The classification "{name}" was not found'
|
||
|
|
return classifications[0]
|
||
|
|
|
||
|
|
|
||
|
|
@step(u'The classification {name} must be used')
|
||
|
|
def step_impl(context, name):
|
||
|
|
get_classification(name)
|
||
|
|
|
||
|
|
|
||
|
|
@step(u'The classification {name} is published by {source}')
|
||
|
|
def step_impl(context, name, source):
|
||
|
|
assert_attribute(get_classification(name), 'Source', source)
|
||
|
|
|
||
|
|
|
||
|
|
@step(u'The classification {name} is the edition {edition} on {edition_date}')
|
||
|
|
def step_impl(context, name, edition, edition_date):
|
||
|
|
element = get_classification(name)
|
||
|
|
assert_attribute(element, 'Edition', edition)
|
||
|
|
assert_attribute(element, 'EditionDate', edition_date)
|
||
|
|
|
||
|
|
|
||
|
|
@step(u'The classification {name} has the description "{description}"')
|
||
|
|
def step_impl(context, name, description):
|
||
|
|
assert_attribute(get_classification(name), 'Description', description)
|
||
|
|
|
||
|
|
|
||
|
|
@step(u'The classification {name} is referenced by the website {location}')
|
||
|
|
def step_impl(context, name, location):
|
||
|
|
assert_attribute(get_classification(name), 'Location', location)
|
||
|
|
|
||
|
|
|
||
|
|
@step(u'The classification {name} has a hierarchy denoted by the tokens {tokens}')
|
||
|
|
def step_impl(context, name, tokens):
|
||
|
|
try:
|
||
|
|
tokens = json.loads(tokens)
|
||
|
|
except:
|
||
|
|
assert False, f'Tokens {tokens} are not specified as a JSON list'
|
||
|
|
assert_attribute(get_classification(name), 'ReferenceTokens', tokens)
|
||
|
|
|
||
|
|
|
||
|
|
@step(u'The element {guid} is classified as a "{identification}" with name "{reference_name}"')
|
||
|
|
def step_impl(context, guid, identification, reference_name):
|
||
|
|
element = IfcFile.by_guid(guid)
|
||
|
|
if not hasattr(element, 'HasAssociations') or not element.HasAssociations:
|
||
|
|
assert False, f'The element {element} has no associations.'
|
||
|
|
references = [a.RelatingClassification for a in element.HasAssociations if a.is_a('IfcRelAssociatesClassification')]
|
||
|
|
if not references:
|
||
|
|
assert False, f'The element {element} has no associated classification references.'
|
||
|
|
is_success = False
|
||
|
|
for reference in references:
|
||
|
|
try:
|
||
|
|
assert_attribute(reference, 'Identification', identification)
|
||
|
|
assert_attribute(reference, 'Name', reference_name)
|
||
|
|
is_success = True
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
if not is_success:
|
||
|
|
assert False, 'No classification references met the requirement for an identification {} and name {} for the element {}. The references we found were: {}'.format(identification, reference_name, element, references)
|