IfcOpenShell/src/ifcbimtester/features/steps/utils.py

import ifcopenshell
import ifcopenshell.util
import ifcopenshell.util.element

class IfcFile(object):
    file = None
    bookmarks = {}

    @classmethod
    def load(cls, path=None):
        cls.file = ifcopenshell.open(path)

    @classmethod
    def get(cls):
        if not cls.file:
            assert False, 'No file was loaded, so this requirement cannot be checked'
        return cls.file

    @classmethod
    def by_guid(cls, guid):
        try:
            return cls.get().by_guid(guid)
        except:
            assert False, 'An element with the ID {} could not be found.'.format(guid)

def assert_number(number):
    try:
        return float(number)
    except ValueError:
        assert False, 'A number should be specified, not {}'.format(number)

def assert_attribute(element, name, value):
    if value == 'NULL':
        value = None
    if not hasattr(element, name):
        assert False, 'The element {} does not have the attribute {}'.format(element, name)
    actual_value = getattr(element, name)
    assert actual_value == value, 'We expected a value of "{}" but instead got "{}" for the element {}'.format(value, actual_value, element)

def assert_pset(element, pset_name, prop_name=None, value=None):
    if value == 'NULL':
        value = None
    psets = ifcopenshell.util.element.get_psets(site)
    if pset_name not in psets:
        assert False, 'The element {} does not have a property set named {}'.format(element, pset_name)
    if prop_name is None:
        return psets[pset_name]
    if prop_name not in psets[pset_name]:
        assert False, 'The element {} does not have a property named "{}" in the pset "{}"'.format(element, prop_name, pset_name)
    if value is None:
        return psets[pset_name][prop_name]
    actual_value = psets[pset_name][prop_name]
    assert actual_value == value, 'We expected a value of "{}" but instead got "{}" for the element {}'.format(value, actual_value, element)