Python assert_datarray_equalの例

コード例 #1

ファイル: test_utils.py プロジェクト: ivanov/datarray

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)

コード例 #2

ファイル: test_utils.py プロジェクト: BIDS/datarray

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)        

コード例 #3

ファイル: test_bugfixes.py プロジェクト: cfarrow/datarray

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)))

コード例 #4

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)))

コード例 #5

ファイル: test_bugfixes.py プロジェクト: cfarrow/datarray

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)

コード例 #6

ファイル: test_bugfixes.py プロジェクト: miketrumpis/datarray

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'))

コード例 #7

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'))

コード例 #8

ファイル: test_bugfixes.py プロジェクト: cfarrow/datarray

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)