Python _remove_rows_by_count Examples

Example #1

File: test_synthesis.py Project: UDST/spandex

def test_remove_rows_by_count_noop(df, count):
    amount = 0

    result = syn._remove_rows_by_count(df, amount, count)

    assert result[count].sum() == df[count].sum()
    pdt.assert_frame_equal(result, df)

Example #2

File: test_synthesis.py Project: terratenney/spandex

def test_remove_rows_by_count_noop(df, count):
    amount = 0

    result = syn._remove_rows_by_count(df, amount, count)

    assert result[count].sum() == df[count].sum()
    pdt.assert_frame_equal(result, df)

Example #3

File: test_synthesis.py Project: UDST/spandex

def test_remove_rows_by_count(df, count, alloc_id):
    amount = 10

    result = syn._remove_rows_by_count(df, amount, count)

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

Example #4

File: test_synthesis.py Project: terratenney/spandex

def test_remove_rows_by_count(df, count, alloc_id):
    amount = 10

    result = syn._remove_rows_by_count(df, amount, count)

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