Add tests to calculate learning_rate in PA1/PA2