Implement `set()` method for CSRMatrix and add tests