def test_assert_datarray_equal(): # Test assert_datarray_equal x = DataArray([1, 2]) y = DataArray([1, 2]) assert_datarray_equal(x, y, "Should not raise assertion") y = DataArray([1, 3]) assert_raises(AssertionError, assert_datarray_equal, x, y) y = DataArray([1, 2, 3]) assert_raises(AssertionError, assert_datarray_equal, x, y) y = DataArray([1, 2], 'a') assert_raises(AssertionError, assert_datarray_equal, x, y) y = DataArray([1, 2], [('a', ['a', 'b'])]) assert_raises(AssertionError, assert_datarray_equal, x, y) x = DataArray([1, 2], 'a') y = DataArray([1, 2], 'a') assert_datarray_equal(x, y, "Should not raise assertion") y = DataArray([1, 2], 'b') assert_raises(AssertionError, assert_datarray_equal, x, y) y = DataArray([1, 2], [('b', ['a', 'b'])]) assert_raises(AssertionError, assert_datarray_equal, x, y) x = DataArray([1, 2], 'a') y = DataArray([1, 2], [('a', None)]) assert_datarray_equal(x, y, "Should not raise assertion") x = DataArray([[1, 2], [3, 4]], [('ax1', ['a', 'b']), ('ax2', ['a', 'b'])]) y = DataArray([[1, 2], [3, 4]], [('ax1', ['a', 'b']), ('ax2', ['a', 'b'])]) assert_datarray_equal(x, y, "Should not raise assertion") y = DataArray([[1, 2], [3, 4]], [('ax1', ['X', 'b']), ('ax2', ['a', 'b'])]) assert_raises(AssertionError, assert_datarray_equal, x, y) y = DataArray([[1, 2], [3, 4]], [('ax1', ['a', 'b']), ('ax2', None)]) assert_raises(AssertionError, assert_datarray_equal, x, y) y = DataArray([[9, 2], [3, 4]], [('ax1', ['a', 'b']), ('ax2', ['a', 'b'])]) assert_raises(AssertionError, assert_datarray_equal, x, y) x = DataArray([1, np.nan]) y = DataArray([1, np.nan]) assert_datarray_equal(x, y, "Should not raise assertion") x = DataArray([1, 2], 'a') y = 1 assert_raises(AssertionError, assert_datarray_equal, x, y) y = np.array([1, 2]) assert_raises(AssertionError, assert_datarray_equal, x, y) x = 1 y = 2 assert_raises(AssertionError, assert_datarray_equal, x, y) x = np.array([1]) y = np.array([2]) assert_raises(AssertionError, assert_datarray_equal, x, y)
def test_bug38(): "Bug 38: DataArray.__repr__ should parse as a single entity" # Calling repr() on an ndarray prepends array (instead of np.array) array = np.array arys = ( DataArray(np.random.randint(0, 10000, size=(1,2,3,4,5)), 'abcde'), DataArray(np.random.randint(0, 10000, size=(3,3,3))), # Try with missing axes DataArray(np.random.randint(0, 10000, (2,4,5,6)), # Try with ticks ('a', ('b', ('b1','b2','b3','b4')), 'c', 'd')), ) for A in arys: print A assert_datarray_equal(A, eval(repr(A)))
def test_bug38(): "Bug 38: DataArray.__repr__ should parse as a single entity" # Calling repr() on an ndarray prepends array (instead of np.array) arys = ( DataArray(np.random.randint(0, 10000, size=(1,2,3,4,5)), 'abcde'), DataArray(np.random.randint(0, 10000, size=(3,3,3))), # Try with missing axes DataArray(np.random.randint(0, 10000, (2,4,5,6)), # Try with ticks ('a', ('b', ('b1','b2','b3','b4')), 'c', 'd')), ) # Load `array` into namespace for `eval` array = np.array for A in arys: assert_datarray_equal(A, eval(repr(A)))
def test_bug22(): "Bug 22: DataArray does not accepting array as ticks" A = DataArray([1, 2], [('time', ['a', 'b'])]) B = DataArray([1, 2], [('time', np.array(['a', 'b']))]) assert_datarray_equal(A, B)
def test_bug45(): "Bug 45" A = DataArray([1,2,3], 'a'); B = DataArray([2,3,4], 'b'); C = np.outer(A,B) assert_datarray_equal(C,DataArray(C, 'ab'))
def test_bug45(): "Bug 45: Support for np.outer()" A = DataArray([1, 2, 3], 'a') B = DataArray([2, 3, 4], 'b') C = np.outer(A, B) assert_datarray_equal(C, DataArray(C, 'ab'))