Skip to content

Commit

Permalink
Renaming indicator in class param
Browse files Browse the repository at this point in the history
  • Loading branch information
davidmunoznovoa committed Oct 1, 2024
1 parent 6b798ce commit 12d5321
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions mesures/cupsdat.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down
4 changes: 2 additions & 2 deletions spec/generation_files_spec.py
Original file line number Diff line number Diff line change
Expand Up @@ -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' \
Expand Down

0 comments on commit 12d5321

Please sign in to comment.