IntelPython / dpnp / 12036779784

81%

push

github

Implement `dpnp.compress` and `dpnp_array.compress` method (#2177)

* Implement dpnp.compress

* Add `dpnp_array.compress` method

* Break up `compress` to satisfy pylint

Also disable checks for protected access, as `compress` uses dpctl.tensor private functions

* Unmute third-party tests for `compress`

* Use `get_usm_allocations` in `compress`

* Fix bug where `out` in `compress` is dpnp_array

Also removes an unnecessary check per PR review

* Apply comments per PR review by @antonwolfy

Also fix a typo when `condition` is not an array

* Remove branching when `condition` is an array

Also tweaks to docstring

* Add tests for `compress`

* Re-use `_take_index` for `dpnp.take`

Should slightly improve efficiency by escaping an additional copy where `out` is not `None` and flattening of indices

* Change error for incorrect out array dtype to `TypeError`

* Move compress tests into a TestCompress class

* Use NumPy in compress tests

* Add `no_none=True` to `test_compress_condition_all_dtypes`

* Add USM type and SYCL queue tests for `compress`

* More tests for compress added

* Docstring change per PR review

* Integrate test for compute follows data in compress into test_2in_1out

* Add test for `dpnp_array.compress` and add a test for strided inputs to `compress`

* Refactor `test_compress` in test_usm_type.py into `test_2in_1out`

4383 of 10012 branches covered (43.78%)

Branch coverage included in aggregate %.

50 of 50 new or added lines in 2 files covered. (100.0%)

16443 of 20695 relevant lines covered (79.45%)

20615.69 hits per line