def test_works_with_normal_and_quantity_arrays_mixed(self):
a = sp.array([1, 2, 3])
b = sp.array([4, 5]) * pq.m
expected_a = sp.array([[1, 2, 3], [1, 2, 3]])
expected_b = sp.array([[4, 4, 4], [5, 5, 5]]) * pq.m
actual_a, actual_b = spq.meshgrid(a, b)
assert_array_equal(expected_a, actual_a)
assert_array_equal(expected_b, actual_b)
a = sp.array([1, 2, 3]) * pq.s
b = sp.array([4, 5])
expected_a = sp.array([[1, 2, 3], [1, 2, 3]]) * pq.s
expected_b = sp.array([[4, 4, 4], [5, 5, 5]])
actual_a, actual_b = spq.meshgrid(a, b)
assert_array_equal(expected_a, actual_a)
assert_array_equal(expected_b, actual_b)
def test_works_with_normal_and_quantity_arrays_mixed(self):
a = sp.array([1, 2, 3])
b = sp.array([4, 5]) * pq.m
expected_a = sp.array([[1, 2, 3], [1, 2, 3]])
expected_b = sp.array([[4, 4, 4], [5, 5, 5]]) * pq.m
actual_a, actual_b = spq.meshgrid(a, b)
assert_array_equal(expected_a, actual_a)
assert_array_equal(expected_b, actual_b)
a = sp.array([1, 2, 3]) * pq.s
b = sp.array([4, 5])
expected_a = sp.array([[1, 2, 3], [1, 2, 3]]) * pq.s
expected_b = sp.array([[4, 4, 4], [5, 5, 5]])
actual_a, actual_b = spq.meshgrid(a, b)
assert_array_equal(expected_a, actual_a)
assert_array_equal(expected_b, actual_b)
def test_works_with_normal_arrays(self):
a = sp.array([1, 2, 3])
b = sp.array([4, 5])
expected_a = sp.array([[1, 2, 3], [1, 2, 3]])
expected_b = sp.array([[4, 4, 4], [5, 5, 5]])
actual_a, actual_b = spq.meshgrid(a, b)
assert_array_equal(expected_a, actual_a)
assert_array_equal(expected_b, actual_b)
def test_works_with_normal_arrays(self):
a = sp.array([1, 2, 3])
b = sp.array([4, 5])
expected_a = sp.array([[1, 2, 3], [1, 2, 3]])
expected_b = sp.array([[4, 4, 4], [5, 5, 5]])
actual_a, actual_b = spq.meshgrid(a, b)
assert_array_equal(expected_a, actual_a)
assert_array_equal(expected_b, actual_b)
