コード例 #1
0
ファイル: test_synthesis.py プロジェクト: UDST/spandex 
def test_add_rows_noop(seed, df):
    num = 0

    result = syn._remove_rows(df, num)

    assert len(result) == len(df) + num
    pdt.assert_frame_equal(result, df)
コード例 #2
0
ファイル: test_synthesis.py プロジェクト: terratenney/spandex 
def test_add_rows_noop(seed, df):
    num = 0

    result = syn._remove_rows(df, num)

    assert len(result) == len(df) + num
    pdt.assert_frame_equal(result, df)
コード例 #3
0
ファイル: test_synthesis.py プロジェクト: UDST/spandex 
def test_remove_rows(seed, df, alloc_id, count):
    num = 2

    result = syn._remove_rows(df, num)

    assert len(result) == len(df) - num
    pdt.assert_frame_equal(
        result,
        pd.DataFrame(
            {alloc_id: ['b', 'c', 'b'],
             count: [2, 3, 4]},
            index=[1, 2, 3]))
コード例 #4
0
ファイル: test_synthesis.py プロジェクト: terratenney/spandex 
def test_remove_rows(seed, df, alloc_id, count):
    num = 2

    result = syn._remove_rows(df, num)

    assert len(result) == len(df) - num
    pdt.assert_frame_equal(
        result,
        pd.DataFrame({
            alloc_id: ['b', 'c', 'b'],
            count: [2, 3, 4]
        },
                     index=[1, 2, 3]))