def main(): matches = extract_data('matches.csv') deliveries = extract_data('deliveries.csv') compute_and_plot_number_of_matches_played_per_year(matches) compute_and_plot_matches_won_by_each_team_over_all_seasons(matches) compute_and_plot_extra_runs_given_by_each_team(matches, deliveries) compute_and_plot_most_economic_bowlers_of_2015(matches, deliveries) compute_centuries_data_over_all_years(matches, deliveries)
import unittest from file_extraction import extract_data from extras_of_year_2016 import compute_extra_runs delivery_data = extract_data('mock_deliveries_data.csv') class TestExtraRunsConcededPerTeam2016(unittest.TestCase): def test_id_wise_data(self): inputs_and_outputs =\ [((['594', '607', '631', '632', '633', '634', '635'], delivery_data), {'Sunrisers Hyderabad': 2, 'Gujarat Lions': 1, 'Kolkata Knight Riders': 2, 'Delhi Daredevils': 5, 'Royal Challengers Bangalore': 1})] for inputs, expected_output in inputs_and_outputs: output = compute_extra_runs(inputs[0], inputs[1]) self.assertEqual(output, expected_output) if __name__ == '__main__': unittest.main()
import unittest from file_extraction import extract_data from max_centuries import ids_of_each_year mock_data = extract_data('matches.csv') mock_matches = [] for mockdata in mock_data[:5]: mock_matches.append(mockdata) delivery_data = extract_data('deliveries.csv') mock_deliveries = [] for delivery in delivery_data[:15]: mock_deliveries.append(delivery) class TestCenturiesPerYear(unittest.TestCase): def test_ids_of_each_year(self): yr_data = {'2017': {}} op = {'2017': {'1': {}, '2': {}, '3': {}, '4': {}, '5': {}}} inputs_and_outputs = [((yr_data, mock_matches), op)] for inputs, expected_output in inputs_and_outputs: output = ids_of_each_year(inputs[0], inputs[1]) self.assertEqual(output, expected_output) if __name__ == '__main__': unittest.main()
import unittest from file_extraction import extract_data from matches_played_per_year import compute_number_of_matches_played_per_year mock_data = extract_data('matches.csv') mock_matches = [] for mockdata in mock_data[:69]: mock_matches.append(mockdata) class TestMatchesPerSeason(unittest.TestCase): def test_compute_number_of_matches_yearwise(self): inputs_and_outputs = [(mock_matches, {'2017': 59, '2008': 10})] for inputs, expected_output in inputs_and_outputs: output = compute_number_of_matches_played_per_year(inputs) self.assertEqual(output, expected_output) if __name__ == '__main__': unittest.main()