

from __future__ import absolute_import

import pytest

import numpy as np

import dask_distance._utils


def test__bool_cmp_mtx_cnt():
    u = np.array([0, 0, 0, 1, 1, 1, 1, 1, 1, 1], dtype=bool)
    v = np.array([0, 1, 1, 0, 0, 0, 1, 1, 1, 1], dtype=bool)

    uv_cmp_mtx = dask_distance._utils._bool_cmp_mtx_cnt(u, v)

    uv_cmp_mtx_exp = np.array([[1, 2], [3, 4]], dtype=float)

    assert (np.array(uv_cmp_mtx) == uv_cmp_mtx_exp).all()



from __future__ import absolute_import

import pytest

import numpy as np

import dask_distance._utils


@pytest.mark.parametrize("et, u, v", [
    (ValueError, np.zeros((2,), dtype=bool), np.zeros((3,), dtype=bool)),
])
def test__bool_cmp_mtx_cnt_err(et, u, v):
    with pytest.raises(et):
        dask_distance._utils._bool_cmp_mtx_cnt(u, v)


def test__bool_cmp_mtx_cnt():
    u = np.array([0, 0, 0, 1, 1, 1, 1, 1, 1, 1], dtype=bool)
    v = np.array([0, 1, 1, 0, 0, 0, 1, 1, 1, 1], dtype=bool)

    uv_cmp_mtx = dask_distance._utils._bool_cmp_mtx_cnt(u, v)

    uv_cmp_mtx_exp = np.array([[1, 2], [3, 4]], dtype=float)

    assert (np.array(uv_cmp_mtx) == uv_cmp_mtx_exp).all()

Test mismatched array lengths error in utils
Test mismatched array lengths error in utils

Adds a test for `_bool_cmp_mtx_cnt` to make sure that if two arrays are
provided with two different lengths that it will raise a `ValueError`.

  
  
  from __future__ import absolute_import
  
  import pytest
  
  import numpy as np
  
  import dask_distance._utils
+ 
+ 
+ @pytest.mark.parametrize("et, u, v", [
+     (ValueError, np.zeros((2,), dtype=bool), np.zeros((3,), dtype=bool)),
+ ])
+ def test__bool_cmp_mtx_cnt_err(et, u, v):
+     with pytest.raises(et):
+         dask_distance._utils._bool_cmp_mtx_cnt(u, v)
  
  
  def test__bool_cmp_mtx_cnt():
      u = np.array([0, 0, 0, 1, 1, 1, 1, 1, 1, 1], dtype=bool)
      v = np.array([0, 1, 1, 0, 0, 0, 1, 1, 1, 1], dtype=bool)
  
      uv_cmp_mtx = dask_distance._utils._bool_cmp_mtx_cnt(u, v)
  
      uv_cmp_mtx_exp = np.array([[1, 2], [3, 4]], dtype=float)
  
      assert (np.array(uv_cmp_mtx) == uv_cmp_mtx_exp).all()
  
Test mismatched array lengths error in utils
## Code Before:


from __future__ import absolute_import

import pytest

import numpy as np

import dask_distance._utils


def test__bool_cmp_mtx_cnt():
    u = np.array([0, 0, 0, 1, 1, 1, 1, 1, 1, 1], dtype=bool)
    v = np.array([0, 1, 1, 0, 0, 0, 1, 1, 1, 1], dtype=bool)

    uv_cmp_mtx = dask_distance._utils._bool_cmp_mtx_cnt(u, v)

    uv_cmp_mtx_exp = np.array([[1, 2], [3, 4]], dtype=float)

    assert (np.array(uv_cmp_mtx) == uv_cmp_mtx_exp).all()

## Instruction:
Test mismatched array lengths error in utils
## Code After:


from __future__ import absolute_import

import pytest

import numpy as np

import dask_distance._utils


@pytest.mark.parametrize("et, u, v", [
    (ValueError, np.zeros((2,), dtype=bool), np.zeros((3,), dtype=bool)),
])
def test__bool_cmp_mtx_cnt_err(et, u, v):
    with pytest.raises(et):
        dask_distance._utils._bool_cmp_mtx_cnt(u, v)


def test__bool_cmp_mtx_cnt():
    u = np.array([0, 0, 0, 1, 1, 1, 1, 1, 1, 1], dtype=bool)
    v = np.array([0, 1, 1, 0, 0, 0, 1, 1, 1, 1], dtype=bool)

    uv_cmp_mtx = dask_distance._utils._bool_cmp_mtx_cnt(u, v)

    uv_cmp_mtx_exp = np.array([[1, 2], [3, 4]], dtype=float)

    assert (np.array(uv_cmp_mtx) == uv_cmp_mtx_exp).all()
