def execute_command(self):
args = self.parser.parse_args()
d = DataProvider()
data = d.extract_data()
x = CalculationLogic()
if args.task == 't1':
if args.district and args.year:
if args.gender:
print(x.task1(data, args.year, args.district,args.gender))
else:
print(x.task1(data, args.year, args.district))
elif args.task == 't2':
if args.district:
if args.gender:
print(x.task2(data,args.district,args.gender))
else:
print(x.task2(data, args.district))
elif args.task == 't3':
if args.year:
if args.gender:
print(x.task3(data,args.year,args.gender))
else:
print(x.task3(data,args.year))
elif args.task == 't4':
if args.gender:
print(x.task4(data,args.gender))
else:
print(x.task4(data))
elif args.task == 't5':
if args.district and args.district2:
if args.gender:
print(x.task5(data,args.district,args.district2,args.gender))
else:
print(x.task5(data,args.district,args.district2))
import pytest
from calculation_logic import CalculationLogic
from data_provider import DataProvider
d = DataProvider()
data = d.extract_data()
x = CalculationLogic()
def test_task1():
assert x.task1(data, 2010, 'Pomorskie') == 10481.5
def test_task1_only_women():
assert x.task1(data, 2010, 'Pomorskie', 'kobiety') == 11798
def test_task1_only_men():
assert x.task1(data, 2010, 'Pomorskie', 'mężczyźni') == 9165
def test_task2():
assert x.task2(data, 'Pomorskie') == {
2010: 81,
2011: 74,
2012: 80,
2013: 80,
2014: 71,
2015: 73,
2016: 79,
2017: 78,
