def test_2d22(self): "la.align() crash with empty larrys (gh #67)" a = larry([[]]) b = larry([[1, 2, 3], [4, 5, 6]]) for j in ('inner', 'outer', 'left', 'right', 'skip'): align(a, a, join=j) align(b, b, join=j) align(a, b, join=j) align(b, a, join=j)
def test_2d9(self): "align 2d test #9" y1 = larry([[1, 2], [3, 4]]) y2 = larry([[1, 2, 5], [3, 4, 6]]) a1, a2 = align(y1, y2, 'skip') d1, d2 = y1.copy(), y2.copy() msg = "align 2d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d21(self): "align 1d test #21" y1 = larry([1, 2, 3]) y2 = larry([1, 2, 3, 4]) a1, a2 = align(y1, y2, join='skip') d1, d2 = y1.copy(), y2.copy() msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d12(self): "align 1d test #12" y1 = larry([1, 2]) y2 = larry([1, 2], [['a', 'b']]) a1, a2 = align(y1, y2, join='outer') d1 = larry([1, 2, nan, nan], [[0, 1, 'a', 'b']], dtype=float) d2 = larry([nan, nan, 1, 2], [[0, 1, 'a', 'b']], dtype=float) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_2d6(self): "align 2d test #6" y1 = larry([[1, 2], [3, 4]]) y2 = larry([[1, 2, 5], [3, 4, 6]]) a1, a2 = align(y1, y2, join=['inner', 'outer']) d1 = larry([[1, 2, nan], [3, 4, nan]]) d2 = larry([[1, 2, 5], [3, 4, 6]]) msg = "align 2d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_2d4(self): "align 2d test #4" y1 = larry([[1, 2], [3, 4]]) y2 = larry([[1, 2, 5], [3, 4, 6]]) a1, a2 = align(y1, y2, join='outer') d1 = larry([[1, 2, nan], [3, 4, nan]], dtype=float) d2 = larry([[1, 2, 5], [3, 4, 6]]) msg = "align 2d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_2d1(self): "align 2d test #1" y1 = larry([[1, 2], [3, 4]]) y2 = larry([[1, 2, 5], [3, 4, 6]]) a1, a2 = align(y1, y2) d1 = larry([[1, 2], [3, 4]]) d2 = larry([[1, 2], [3, 4]]) msg = "align 2d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d16(self): "align 1d test #16" y1 = larry([1, 2]) y2 = larry([1, 2], [['a', 'b']]) a1, a2 = align(y1, y2, join=['left']) d1 = larry([1, 2]) d2 = larry([nan, nan]) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d19(self): "align 1d test #19" y1 = larry([True, False]) y2 = larry([True, False, True]) a1, a2 = align(y1, y2, join='outer') d1 = larry([1, 0, nan], dtype=float) d2 = larry([True, False, True]) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d11(self): "align 1d test #11" y1 = larry([1, 2]) y2 = larry([1, 2], [['a', 'b']]) a1, a2 = align(y1, y2, join='inner') d1 = larry([], dtype=int) d2 = larry([], dtype=int) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d6(self): "align 1d test #6" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) a1, a2 = align(y1, y2, join='left') d1 = larry([1, 2]) d2 = larry([1, 2]) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d8(self): "align 1d test #8" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) a1, a2 = align(y1, y2, join=['right']) d1 = larry([1, 2, nan], dtype=float) d2 = larry([1, 2, 3]) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d12(self): "align 1d test #12" y1 = larry([1, 2]) y2 = larry([1, 2], [[3, 4]]) a1, a2 = align(y1, y2, join='outer') d1 = larry([1, 2, nan, nan], [[0, 1, 3, 4]], dtype=float) d2 = larry([nan, nan, 1, 2], [[0, 1, 3, 4]], dtype=float) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d18(self): "align 1d test #18" y1 = larry([1, 2]) y2 = larry([1, 2]) a1, a2 = align(y1, y2, cast=False) d1 = larry([1, 2]) d2 = larry([1, 2]) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_2d8(self): "align 2d test #8" y1 = larry([[1, 2], [3, 4]]) y2 = larry([[1, 2], [3, 4]]) a1, a2 = align(y1, y2, cast=False) d1 = larry([[1, 2], [3, 4]]) d2 = larry([[1, 2], [3, 4]]) msg = "align 2d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_2d10(self): "align 2d test #10" y1 = larry([[1, 2], [3, 4], [7, 8]]) y2 = larry([[1, 2, 5], [3, 4, 6]]) a1, a2 = align(y1, y2, ['inner', 'skip']) d1 = larry([[1, 2], [3, 4]]) d2 = larry([[1, 2, 5], [3, 4, 6]]) msg = "align 2d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_2d11(self): "align 2d test #11" y1 = larry([[0.1, 0.2], [0.3, 0.4], [0.7, 0.8]]) y2 = larry([[0.1, 0.2, 0.5], [0.3, 0.4, 0.6]]) a1, a2 = align(y1, y2, ['skip', 'outer']) d1 = larry([[0.1, 0.2, np.nan], [0.3, 0.4, np.nan], [0.7, 0.8, np.nan]]) d2 = larry([[0.1, 0.2, 0.5], [0.3, 0.4, 0.6]]) msg = "align 2d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d20(self): "align 1d test #20" d = datetime.date y1 = larry([d(2011, 1, 1), d(2011, 1, 2)]) y2 = larry([d(2011, 1, 3), d(2011, 1, 4), d(2011, 1, 5)]) a1, a2 = align(y1, y2, join='outer') d1 = larry([d(2011, 1, 1), d(2011, 1, 2), None]) d2 = larry([d(2011, 1, 3), d(2011, 1, 4), d(2011, 1, 5)]) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)
def test_1d20(self): "align 1d test #20" d = datetime.date y1 = larry([d(2011,1,1), d(2011,1,2)]) y2 = larry([d(2011,1,3), d(2011,1,4), d(2011,1,5)]) a1, a2 = align(y1, y2, join='outer') d1 = larry([d(2011,1,1), d(2011,1,2), None]) d2 = larry([d(2011,1,3), d(2011,1,4), d(2011,1,5)]) msg = "align 1d fail on %s larry" ale(a1, d1, msg % 'left', original=y1) ale(a2, d2, msg % 'right', original=y2)