Skip to content

Commit 244f97e

Browse files
finozzifafinozzifa
committed
code: shorten the conftest.py
1 parent dd2ea29 commit 244f97e

File tree

2 files changed

+93
-173
lines changed

2 files changed

+93
-173
lines changed

‎test/conftest.py‎

Lines changed: 92 additions & 172 deletions
Original file line numberDiff line numberDiff line change
@@ -203,179 +203,99 @@ def mock_output_data():
203203
"""
204204

205205
def mock_output(source):
206+
unit_list = (
207+
["MW", "EUR"]
208+
+ ["/"] * 3
209+
+ ["W"]
210+
+ ["EUR"] * 7
211+
+ ["r/MW"]
212+
+ ["r/MWh"] * 2
213+
+ ["/year"]
214+
+ ["EUR"] * 5
215+
+ [
216+
"MW_e",
217+
"EUR/MW_e",
218+
"MW_th",
219+
"MW_th",
220+
"MWh_th",
221+
"MWh_th",
222+
"MWh_FT",
223+
"MW_FT",
224+
"MW_MeOH",
225+
"MW_FT/year",
226+
"MW_MeOH/year",
227+
"MWh_FT",
228+
"MWh_MeOH",
229+
"MW_CH4/year",
230+
"MWh_CH4",
231+
"MW_CH4",
232+
"EUR/MWh_e",
233+
"EUR/MW_eh",
234+
"MWh_th/MWh_el",
235+
"MWh_FT/MWh_H2",
236+
"MW_NH3",
237+
"MW_NH3",
238+
"MWh_NH3",
239+
"EUR/MW/year",
240+
"EUR/MW",
241+
"EUR/MW/year",
242+
"EUR/MWh",
243+
"MW",
244+
]
245+
)
246+
206247
if source == "dea":
207-
return pd.DataFrame(
208-
{
209-
"technology": ["random_tech"] * 50
210-
+ [
211-
"central air-sourced heat pump",
212-
"central geothermal-sourced heat pump",
213-
"central gas boiler",
214-
"central resistive heater",
215-
"decentral air-sourced heat pump",
216-
"decentral gas boiler",
217-
"decentral ground-sourced heat pump",
218-
]
219-
+ ["fuel cell"] * 4,
220-
"unit": [
221-
"MW",
222-
"EUR",
223-
"/",
224-
"/",
225-
"/",
226-
"W",
227-
"EUR",
228-
"EUR",
229-
"EUR",
230-
"EUR",
231-
"EUR",
232-
"EUR",
233-
"EUR",
234-
"r/MW",
235-
"r/MWh",
236-
"r/MWh",
237-
"/year",
238-
"EUR",
239-
"EUR",
240-
"EUR",
241-
"EUR",
242-
"EUR",
243-
"MW_e",
244-
"EUR/MW_e",
245-
"MW_th",
246-
"MW_th",
247-
"MWh_th",
248-
"MWh_th",
249-
"MWh_FT",
250-
"MW_FT",
251-
"MW_MeOH",
252-
"MW_FT/year",
253-
"MW_MeOH/year",
254-
"MWh_FT",
255-
"MWh_MeOH",
256-
"MW_CH4/year",
257-
"MWh_CH4",
258-
"MW_CH4",
259-
"EUR/MWh_e",
260-
"EUR/MW_eh",
261-
"MWh_th/MWh_el",
262-
"MWh_FT/MWh_H2",
263-
"MW_NH3",
264-
"MW_NH3",
265-
"MWh_NH3",
266-
"EUR/MW/year",
267-
"EUR/MW",
268-
"EUR/MW/year",
269-
"EUR/MWh",
270-
"MW",
271-
"EUR/MW_th",
272-
"EUR/MW_th/year",
273-
"EUR/MWh_th",
274-
"MW_th",
275-
"EUR/MW_th",
276-
"EUR/MW_th/year",
277-
"EUR/MWh_th",
278-
"EUR/MW_e",
279-
"EUR/MW_e/year",
280-
"EUR/MWh_e",
281-
"MW_e",
282-
],
283-
"value": [1.0] * 6
284-
+ [
285-
0.8917822267802202,
286-
1.177107611177814,
287-
]
288-
+ [1.0] * 7
289-
+ [3.6]
290-
+ [1.0] * 45,
291-
},
292-
).set_index(["technology"])
248+
unit_list += [
249+
"EUR/MW_th",
250+
"EUR/MW_th/year",
251+
"EUR/MWh_th",
252+
"MW_th",
253+
"EUR/MW_th",
254+
"EUR/MW_th/year",
255+
"EUR/MWh_th",
256+
"EUR/MW_e",
257+
"EUR/MW_e/year",
258+
"EUR/MWh_e",
259+
"MW_e",
260+
]
293261
else:
294-
return pd.DataFrame(
295-
{
296-
"technology": ["random_tech"] * 50
297-
+ [
298-
"central air-sourced heat pump",
299-
"central geothermal-sourced heat pump",
300-
"central gas boiler",
301-
"central resistive heater",
302-
"decentral air-sourced heat pump",
303-
"decentral gas boiler",
304-
"decentral ground-sourced heat pump",
305-
]
306-
+ ["fuel cell"] * 4,
307-
"unit": [
308-
"MW",
309-
"EUR",
310-
"/",
311-
"/",
312-
"/",
313-
"W",
314-
"EUR",
315-
"EUR",
316-
"EUR",
317-
"EUR",
318-
"EUR",
319-
"EUR",
320-
"EUR",
321-
"r/MW",
322-
"r/MWh",
323-
"r/MWh",
324-
"/year",
325-
"EUR",
326-
"EUR",
327-
"EUR",
328-
"EUR",
329-
"EUR",
330-
"MW_e",
331-
"EUR/MW_e",
332-
"MW_th",
333-
"MW_th",
334-
"MWh_th",
335-
"MWh_th",
336-
"MWh_FT",
337-
"MW_FT",
338-
"MW_MeOH",
339-
"MW_FT/year",
340-
"MW_MeOH/year",
341-
"MWh_FT",
342-
"MWh_MeOH",
343-
"MW_CH4/year",
344-
"MWh_CH4",
345-
"MW_CH4",
346-
"EUR/MWh_e",
347-
"EUR/MW_eh",
348-
"MWh_th/MWh_el",
349-
"MWh_FT/MWh_H2",
350-
"MW_NH3",
351-
"MW_NH3",
352-
"MWh_NH3",
353-
"EUR/MW/year",
354-
"EUR/MW",
355-
"EUR/MW/year",
356-
"EUR/MWh",
357-
"MW",
358-
"EUR/MW",
359-
"EUR/MW/year",
360-
"EUR/MWh",
361-
"MW",
362-
"EUR/MW",
363-
"EUR/MW/year",
364-
"EUR/MWh",
365-
"EUR/MW",
366-
"EUR/MW/year",
367-
"EUR/MWh",
368-
"MW",
369-
],
370-
"value": [1.0] * 6
371-
+ [
372-
0.8917822267802202,
373-
1.177107611177814,
374-
]
375-
+ [1.0] * 7
376-
+ [3.6]
377-
+ [1.0] * 45,
378-
},
379-
).set_index(["technology"])
262+
unit_list += [
263+
"EUR/MW",
264+
"EUR/MW/year",
265+
"EUR/MWh",
266+
"MW",
267+
"EUR/MW",
268+
"EUR/MW/year",
269+
"EUR/MWh",
270+
"EUR/MW",
271+
"EUR/MW/year",
272+
"EUR/MWh",
273+
"MW",
274+
]
275+
276+
return pd.DataFrame(
277+
{
278+
"technology": ["random_tech"] * 50
279+
+ [
280+
"central air-sourced heat pump",
281+
"central geothermal-sourced heat pump",
282+
"central gas boiler",
283+
"central resistive heater",
284+
"decentral air-sourced heat pump",
285+
"decentral gas boiler",
286+
"decentral ground-sourced heat pump",
287+
]
288+
+ ["fuel cell"] * 4,
289+
"unit": unit_list,
290+
"value": [1.0] * 6
291+
+ [
292+
0.8917822267802202,
293+
1.177107611177814,
294+
]
295+
+ [1.0] * 7
296+
+ [3.6]
297+
+ [1.0] * 45,
298+
},
299+
).set_index(["technology"])
380300

381301
return mock_output

‎test/test_compile_cost_assumptions.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,8 @@ def test_clean_up_units(mock_input_data, mock_output_data, source):
6060
output_df = clean_up_units(
6161
mock_input_data.copy(deep=True), value_column="value", source=source
6262
)
63-
print(mock_input_data.shape, output_df.shape, expected_df.shape)
6463
comparison_df = output_df.compare(expected_df)
64+
print(comparison_df)
6565
assert comparison_df.empty
6666

6767

0 commit comments

Comments
 (0)