def _get_simple_text() -> StructuredText:
row1 = Row([_cell('AA'), _cell('BB'), _cell('CC')], True)
row2 = Row([_cell('DD', rs=2), _cell('EE', cs=2)], False)
row3 = Row([_cell('FF'), _cell('GG')], False)
table = Table([row1, row2, row3])
alineas = [
_text('Alinea 1'),
_text('Alinea 2'),
EnrichedString('', table=table)
]
sections = [StructuredText(_text('Title 2'), [], [], None, None, None)]
text = StructuredText(_text('Title'), alineas, sections, None, None, None)
return text
def compute_detection_performance(dataset: List[_LabelizedText], ontology: TopicOntology):
all_labels = [label for _, label in dataset]
texts = [
(titles, StructuredText(EnrichedString(''), outer_alineas, [], None)) for (titles, outer_alineas), _ in dataset
]
for topic, texts_ in _missing_topics(all_labels, texts, ontology).items():
_pretty_print(topic, texts_)
predicted_labels = [_extract_topics(text, titles, ontology) for titles, text in texts]
print(Counter([exp == pred for exp, pred in zip(all_labels, predicted_labels) if exp]))
def _node_section() -> StructuredText:
return StructuredText(
_str('abc'),
[_str('abc'), _enriched_string_table()],
[_leaf_section()],
None,
Reference('ref', 'name'),
Annotations(None),
)
def test_add_topics():
sub_section_1 = StructuredText(EnrichedString('Section 1.1'), [], [], None)
section_1 = StructuredText(EnrichedString('Section 1'), [],
[sub_section_1], None)
section_2 = StructuredText(EnrichedString('Section 2'), [], [], None)
am = ArreteMinisteriel(EnrichedString('arrete du 10/10/10'),
[section_1, section_2], [],
None,
id='FAKE_ID')
am_with_topics = add_topics(
am, {
(0, ): TopicName.INCENDIE,
(0, 0): TopicName.INCENDIE,
(1, ): TopicName.BRUIT_VIBRATIONS
})
sections = am_with_topics.sections
assert sections[0].annotations.topic == TopicName.INCENDIE # type: ignore
assert sections[0].sections[
0].annotations.topic == TopicName.INCENDIE # type: ignore
assert sections[
1].annotations.topic == TopicName.BRUIT_VIBRATIONS # type: ignore
def test_extract_text_lines():
assert _get_simple_text().text_lines() == [
'AM',
'alinea',
'foo',
'# Section 1',
'## Section 1.1',
'# Section 2',
'bar',
]
assert StructuredText(EnrichedString(' A'), [], [], None).text_lines() == ['A']
assert StructuredText(EnrichedString(' A'), [EnrichedString('')], [], None).text_lines() == ['A', '']
assert _TEXT_A.text_lines() == [
'6. Schématisation des différents types de joints mentionnés :',
'Vous pouvez consulter les schémas dans le',
'JO',
'n° 265 du 16/11/2010 texte numéro 21',
]
assert _TEXT_B.text_lines() == [
'6. Schématisation des différents types de joints mentionnés :',
'Vous pouvez consulter les schémas dans le',
'JO n° 265 du 16/11/2010 texte numéro 21',
]
def _build_am() -> ArreteMinisteriel:
subsection1 = StructuredText(EnrichedString('A'), [], [], None)
subsection2 = StructuredText(EnrichedString('A'), [], [], None)
sections = [
StructuredText(EnrichedString('Art. 1'), [], [subsection1], None),
StructuredText(EnrichedString('Art. 2'), [], [subsection2], None),
StructuredText(EnrichedString('Conditions d\'application'), [], [],
None),
StructuredText(EnrichedString('Art. 3'), [], [], None),
]
return ArreteMinisteriel(EnrichedString('arrete du 10/10/10'),
sections, [],
None,
id='FAKE_ID')
def test_check_parametrization_consistency():
date_ = ParameterEnum.DATE_INSTALLATION.value
dt_1 = datetime.now()
dt_2 = dt_1 + timedelta(days=1)
cd_1 = Range(date_, dt_1, dt_2)
cd_3 = Littler(date_, dt_1)
cd_4 = Greater(date_, dt_2)
cd_5 = Littler(date_, dt_2)
cd_6 = Greater(date_, dt_1)
new_text = StructuredText(_str('Art. 2'), [_str('version modifiée')], [],
None)
Parametrization(
[_NAC('', None, cd_1), _NAC('', None, cd_3)],
[AlternativeSection('', new_text, cd_4)], []).check_consistency()
Parametrization([_NAC('', None, cd_3)], [], []).check_consistency()
with pytest.raises(ParametrizationError):
Parametrization(
[_NAC('', None, cd_1), _NAC('', None, cd_5)],
[AlternativeSection('', new_text, cd_6)], []).check_consistency()
with pytest.raises(ParametrizationError):
Parametrization(
[_NAC('', None, cd_1), _NAC('', None, cd_1)], [],
[]).check_consistency()
def _leaf_section() -> StructuredText:
app = Applicability(True, True, ['beware'], StructuredText(_str('abc'), [], [], None))
annotations = Annotations(TopicName.AIR_ODEURS)
return StructuredText(_str('abc'), [_str('abc')], [], app, Reference('ref', 'name'), annotations)
def _get_simple_text() -> StructuredText:
sub_section_1 = StructuredText(_str('Section 1.1'), [], [], None)
section_1 = StructuredText(_str('Section 1'), [], [sub_section_1], None)
section_2 = StructuredText(_str('Section 2'), [_str('bar')], [], None)
return StructuredText(_str('AM '), [_str('alinea'), _str('foo')], [section_1, section_2], None)
assert not _is_probably_cid('JORFTEX')
def _get_simple_text() -> StructuredText:
sub_section_1 = StructuredText(_str('Section 1.1'), [], [], None)
section_1 = StructuredText(_str('Section 1'), [], [sub_section_1], None)
section_2 = StructuredText(_str('Section 2'), [_str('bar')], [], None)
return StructuredText(_str('AM '), [_str('alinea'), _str('foo')], [section_1, section_2], None)
_TEXT_A = StructuredText(
title=EnrichedString(text='6. Schématisation des différents types de joints mentionnés :'),
outer_alineas=[
EnrichedString(text='Vous pouvez consulter les schémas dans le'),
EnrichedString(text='JO\nn° 265 du 16/11/2010 texte numéro 21'),
],
sections=[],
applicability=None,
reference=Reference('ref', 'name'),
annotations=None,
id='0bEB0b14A96f',
)
_TEXT_B = StructuredText(
title=EnrichedString(text='6. Schématisation des différents types de joints mentionnés :'),
outer_alineas=[
EnrichedString(text='Vous pouvez consulter les schémas dans le'),
EnrichedString(text='JO n° 265 du 16/11/2010 texte numéro 21'),
],
sections=[],
applicability=None,
reference=None,
annotations=None,
def test_add_references():
sub_sub_sections = [
StructuredText(estr('1.1. azeaze'), [], [], None, None),
StructuredText(estr('1. 2. azeaze'), [], [], None, None),
]
sub_sections = [
StructuredText(estr('1. azeaze'), [], sub_sub_sections, None, None)
]
sections = [
StructuredText(estr('Article 1. efzefz'), [], sub_sections, None),
StructuredText(estr('2. zefez'), [], [], None),
StructuredText(estr('A. zefze'), [], [], None),
StructuredText(estr('a) zefze'), [], [], None),
StructuredText(estr('V. zefze'), [], [], None),
StructuredText(estr('ANNEXE I zefze'), [], [], None),
StructuredText(estr('Article 18.1'), [], [], None),
StructuredText(estr('Article 1'), [], [], None),
]
am = ArreteMinisteriel(estr('Arrete du 10/10/10'),
sections, [],
None,
id='FAKE_ID')
am_with_references = add_references(am)
secs = am_with_references.sections
assert ensure_not_none(secs[0].reference).nb == 'Article 1.'
assert ensure_not_none(
secs[0].sections[0].reference).nb == 'Article 1. > 1.'
assert ensure_not_none(
secs[0].sections[0].sections[0].reference).nb == 'Article 1. > 1.1.'
assert ensure_not_none(
secs[0].sections[0].sections[1].reference).nb == 'Article 1. > 1.2.'
assert ensure_not_none(secs[1].reference).nb == '2.'
assert ensure_not_none(secs[2].reference).nb == 'A.'
assert ensure_not_none(secs[3].reference).nb == 'a)'
assert ensure_not_none(secs[4].reference).nb == 'V.'
assert ensure_not_none(secs[5].reference).nb == 'Annexe I'
assert ensure_not_none(secs[6].reference).nb == 'Article 18.1'
assert ensure_not_none(secs[7].reference).nb == 'Article 1'
from envinorma.models.text_elements import EnrichedString
from envinorma.parametrization.combinations import _generate_options_dicts, generate_exhaustive_combinations
from envinorma.parametrization.models.parametrization import (
AlternativeSection,
AMWarning,
InapplicableSection,
Parametrization,
extract_conditions_from_parametrization,
)
_DATE = Parameter(id='date-d-installation', type=ParameterType.DATE)
_NEW_TEXT = StructuredText(
title=EnrichedString(text='Article 2.1'),
outer_alineas=[EnrichedString(text='Contenu nouveau')],
sections=[],
applicability=None,
reference=None,
annotations=None,
id='d16d0fE7C7fc',
)
_PARAMETRIZATION = Parametrization(
inapplicable_sections=[
InapplicableSection(
section_id='abcdef',
alineas=None,
condition=AndCondition(
conditions=frozenset([Littler(parameter=_DATE, target=date(2021, 1, 1), strict=True)])
),
)
],
alternative_sections=[