Compute the total loss

Yes. Here is the checklist.

The compute_total_loss_test code is correct and it is just testing your code.