From 12d53215aa679905786ed26c6604384c5f96ff29 Mon Sep 17 00:00:00 2001 From: davidmunoznovoa Date: Tue, 1 Oct 2024 16:43:00 +0200 Subject: [PATCH] Renaming indicator in class param --- mesures/cupsdat.py | 6 +++--- spec/generation_files_spec.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/mesures/cupsdat.py b/mesures/cupsdat.py index 472a2df..b2a5ebb 100644 --- a/mesures/cupsdat.py +++ b/mesures/cupsdat.py @@ -8,17 +8,17 @@ class CUPSDAT(object): - def __init__(self, data, distributor=None, compression='bz2', columns=COLUMNS, include_measure_indicator=False, + def __init__(self, data, distributor=None, compression='bz2', columns=COLUMNS, include_measure_type_indicator=False, version=0): """ :param data: list of dicts or absolute file_path :param distributor: str distributor REE code :param compression: 'bz2', 'gz'... OR False otherwise - :param include_measure_indicator: boolean (indicates if new columns is included) + :param include_measure_type_indicator: boolean (indicates if new columns is included) """ data = DummyKeys(data).data self.columns = columns - if include_measure_indicator: + if include_measure_type_indicator: self.columns = COLUMNS_2024 self.file = self.reader(data) self.generation_date = datetime.now() diff --git a/spec/generation_files_spec.py b/spec/generation_files_spec.py index 7bb336a..ae2a779 100644 --- a/spec/generation_files_spec.py +++ b/spec/generation_files_spec.py @@ -1302,12 +1302,12 @@ def get_sample_reobjecil_data(): with it('has new field indicador_envio_medida if it is specified in function call'): data = SampleData().get_sample_cupsdat_data_isp_2024() - f = CUPSDAT(data, include_measure_indicator=True) + f = CUPSDAT(data, include_measure_type_indicator=True) assert 'indicador_envio_medida' in f.columns with it('has its expected content when indicador_envio_medida is included'): data = SampleData().get_sample_cupsdat_data_isp_2024() - f = CUPSDAT(data, include_measure_indicator=True) + f = CUPSDAT(data, include_measure_type_indicator=True) res = f.writer() expected = 'ES0291000000004444QR1F;CUPS de Demo 01;X0004444;4444;5555;2;' \ 'E2;6A;G0;A;GI;400;400;400;400;400;500;1091;2022/10/01 01;2022/10/17 00;17005;S;27;Q\n' \