Add tests for total_bytes