def test_getitem_07(): "larry.getitem #07" desired = larry([[ 3.0, 4.0], [ 5.0, 6.0]], [[1, 2], [0, 1]]) lar1, lar2 = make_larrys() actual = lar1[np.array([1, 2])] ale(actual, desired, original=lar1)
def test_sortby_12(self): "sortby test #12" lar = rand(0,2) theory = lar.copy() practice = sortby(lar, 0, 1) msg="Sorting empty larry failed" ale(theory, practice, msg=msg)
def test_panel_2(self): "func.panel_2" x = np.ones((4,3)).cumsum(0) - 1 label = [['r1', 'r2', 'r3', 'r4'], ['c1', 'c2', 'c3']] original = larry(x, label) original = original.insertaxis(0, "name") y = original.copy() y.label[0] = ["two"] original = original.merge(y) actual = panel(original) x = np.array([[0,0,0,1,1,1,2,2,2,3,3,3]], dtype=np.float).T x = np.hstack((x,x)) label = [[('r1', 'c1'), ('r1', 'c2'), ('r1', 'c3'), ('r2', 'c1'), ('r2', 'c2'), ('r2', 'c3'), ('r3', 'c1'), ('r3', 'c2'), ('r3', 'c3'), ('r4', 'c1'), ('r4', 'c2'), ('r4', 'c3')], ['name', 'two']] desired = larry(x, label) ale(actual, desired, msg='panel test #2', original=original)
def test_getitem_16(): "larry.getitem #16" desired = larry([[ 1.0, nan]]) lar1, lar2 = make_larrys() idx = np.array([True, False]) actual = lar1[idx,:] ale(actual, desired, original=lar1)
def test_getitem_06(): "larry.getitem #06" desired = larry([[1.0, nan], [3.0, 4.0]]) lar1, lar2 = make_larrys() actual = lar1[0:2,0:2] ale(actual, desired)
def test_panel_1(self): "func.panel_1" original = larry(np.arange(24).reshape(2,3,4)) actual = panel(original) x = np.array([[ 0, 12], [ 4, 16], [ 8, 20], [ 1, 13], [ 5, 17], [ 9, 21], [ 2, 14], [ 6, 18], [10, 22], [ 3, 15], [ 7, 19], [11, 23]]) label = [[(0, 0), (0, 1), (0, 2), (0, 3), (1, 0), (1, 1), (1, 2), (1, 3), (2, 0), (2, 1), (2, 2), (2, 3)], [0, 1]] desired = larry(x, label) ale(actual, desired, msg='panel test #1', original=original)
def test_binaryop_08(self): "binaryop test #08" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = binaryop(np.add, y1, y2, join='outer', missone=0, misstwo=0) desired = larry([2, 4, 3], dtype=float) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_09(self): "binaryop test #09" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = binaryop(np.add, y1, y2, join='outer', misstwo=0) desired = larry([2, 4, nan]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_02(self): "binaryop test #02" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = binaryop(np.add, y1, y2, join='inner') desired = larry([2, 4]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_05(self): "binaryop test #05" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = binaryop(np.add, y1, y2, misstwo=0) desired = larry([2, 4]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, 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_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_stack_1(self): "func.stack_1" y1 = larry([[1, 2], [3, 4]]) y2 = larry([[5, 6], [7, 8]]) actual = stack('union', name1=y1, othername=y2) x = np.array([[[1., 2.], [3., 4.]], [[5., 6.], [7., 8.]]]) label = [['name1', 'othername'], [0, 1], [0, 1]] desired = larry(x, label) ale(actual, desired, msg='stack test #1')
def test_divide_01(self): "divide test #01" y1 = larry([1, 2, nan], [['a', 'b', 'c']]) y2 = larry([1, nan, nan], [['a', 'b', 'dd']]) actual = divide(y1, y2, join='outer', missone=1, misstwo=2) desired = larry([1, 2, 1, 1], [['a', 'b', 'c', 'dd']], dtype=float) msg = "divide failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_subtract_03(self): "subtract test #03" y1 = larry([1, 2, nan], [['a', 'b', 'c']]) y2 = larry([1, nan, nan], [['a', 'b', 'dd']]) actual = subtract(y1, y2, join='outer') desired = larry([0, nan, nan, nan], [['a', 'b', 'c', 'dd']]) msg = "subtract failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_12(self): "binaryop test #12" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = binaryop(np.add, y1, y2, join='left', missone=0, misstwo=0) desired = larry([2, 4]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_subtract_02(self): "subtract test #02" y1 = larry([1, 2, nan], [['a', 'b', 'c']]) y2 = larry([1, nan, nan], [['a', 'b', 'dd']]) actual = subtract(y1, y2, missone=0) desired = larry([0, 2], [['a', 'b']], dtype=float) msg = "subtract failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_20(self): "add test #20" y1 = larry([1, nan, nan, 1]) y2 = larry([1, 1, nan, 1], [[0, 1, 2, 9]]) actual = add(y1, y2, join='outer', missone=0, misstwo=0) desired = larry([2, 1, 0, 1, 1], [[0, 1, 2, 3, 9]], dtype=float) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_19(self): "add test #19" y1 = larry([1, nan, nan, 1]) y2 = larry([1, 1, nan, 1], [[0, 1, 2, 9]]) actual = add(y1, y2, join='outer', missone=0) desired = larry([2, 1, nan, 1, 1], [[0, 1, 2, 3, 9]]) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_17(self): "add test #17" y1 = larry([1, nan, nan, 1]) y2 = larry([1, 1, nan, 1], [[0, 1, 2, 'a']]) actual = add(y1, y2) desired = larry([2, nan, nan], [[0, 1, 2]]) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_16(self): "add test #16" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = add(y1, y2, join='right', misstwo=0) desired = larry([2, 4, nan]) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_12(self): "add test #12" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = add(y1, y2, join='left', missone=0, misstwo=0) desired = larry([2, 4]) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_15(self): "add test #15" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = add(y1, y2, join='right', missone=0, misstwo=0) desired = larry([2, 4, 3], dtype=float) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_subtract_04(self): "subtract test #04" y1 = larry([1, 2, nan], [['a', 'b', 'c']]) y2 = larry([1, nan, nan], [['a', 'b', 'dd']]) actual = subtract(y1, y2, join='outer', missone=0, misstwo=0) desired = larry([0, 2, 0, 0], [['a', 'b', 'c', 'dd']], dtype=float) msg = "subtract failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_10(self): "add test #10" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = add(y1, y2, join='left') desired = larry([2, 4]) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_multiply_01(self): "multiply test #01" y1 = larry([1, 2, nan], [['a', 'b', 'c']]) y2 = larry([1, nan, nan], [['a', 'b', 'dd']]) actual = multiply(y1, y2, join='outer', missone=1, misstwo=2) desired = larry([1, 2, 4, 4], [['a', 'b', 'c', 'dd']], dtype=float) msg = "multiply failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_04(self): "add test #04" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = add(y1, y2, join='inner', missone=0, misstwo=0) desired = larry([2, 4]) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_06(self): "add test #06" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = add(y1, y2, join='outer') desired = larry([2, 4, nan]) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_07(self): "add test #07" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = add(y1, y2, join='outer', missone=0) desired = larry([2, 4, 3], dtype=float) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_add_02(self): "add test #02" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = add(y1, y2, join='inner') desired = larry([2, 4]) msg = "add failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_17(self): "binaryop test #17" y1 = larry([1, nan, nan, 1]) y2 = larry([1, 1, nan, 1], [[0, 1, 2, 'a']]) actual = binaryop(np.add, y1, y2) desired = larry([2, nan, nan], [[0, 1, 2]]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_18(self): "binaryop test #18" y1 = larry([1, nan, nan, 1]) y2 = larry([1, 1, nan, 1], [[0, 1, 2, 9]]) actual = binaryop(np.add, y1, y2, join='outer') desired = larry([2, nan, nan, nan, nan], [[0, 1, 2, 3, 9]]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_21(self): "binaryop test #21" y1 = larry([True, True]) y2 = larry([True, False, True]) actual = binaryop(np.logical_and, y1, y2) desired = larry([True, False]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_20(self): "binaryop test #20" y1 = larry([1, nan, nan, 1]) y2 = larry([1, 1, nan, 1], [[0, 1, 2, 9]]) actual = binaryop(np.add, y1, y2, join='outer', missone=0, misstwo=0) desired = larry([2, 1, 0, 1, 1], [[0, 1, 2, 3, 9]], dtype=float) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_15(self): "binaryop test #15" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = binaryop(np.add, y1, y2, join='right', missone=0, misstwo=0) desired = larry([2, 4, 3], dtype=float) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_16(self): "binaryop test #16" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = binaryop(np.add, y1, y2, join='right', misstwo=0) desired = larry([2, 4, nan]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_binaryop_06(self): "binaryop test #06" y1 = larry([1, 2]) y2 = larry([1, 2, 3]) actual = binaryop(np.add, y1, y2, join='outer') desired = larry([2, 4, nan]) msg = "binaryop failed" ale(actual, desired, msg, original=y1) ale(actual, desired, msg, original=y2)
def test_cov_2(self): "func.cov_2" original = larry([[nan, 2.0, 1.0], [2.0, 3.0, 1.0], [4.0, 1.0, 1.0]]) original = original.demean(1) desired = larry(np.ma.cov(np.ma.fix_invalid(original.x), bias=1).data) actual = cov(original) ale(actual, desired, msg='cov test #2', original=original)
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_panel_1(self): "func.panel_1" original = larry(np.arange(24).reshape(2, 3, 4)) actual = panel(original) x = np.array([[0, 12], [4, 16], [8, 20], [1, 13], [5, 17], [9, 21], [2, 14], [6, 18], [10, 22], [3, 15], [7, 19], [11, 23]]) label = [[(0, 0), (0, 1), (0, 2), (0, 3), (1, 0), (1, 1), (1, 2), (1, 3), (2, 0), (2, 1), (2, 2), (2, 3)], [0, 1]] desired = larry(x, label) ale(actual, desired, msg='panel test #1', original=original)
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_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_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_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_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)