FYI: for this line that compares scalars you could have used `assert_almost_equal`.