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,