Example #1
0
def test_max_level():
    # manual
    assert_equals(('Lapras', 72), hw2_manual.max_level(_pokemon_test))
    assert_equals(('Persian', 40), hw2_manual.max_level(_single))
    # pandas
    assert_equals(('Lapras', 72), hw2_pandas.max_level(_pokemon_test_df))
    assert_equals(('Persian', 40), hw2_pandas.max_level(_single_df))
Example #2
0
def test_max_level():
    '''
    Test max_level funciton.
    '''
    print("test_max_level")
    print('Part0')
    assert_equals(('Lapras', 72), hw2_manual.max_level(parsed_data))
    assert_equals(('Victreebel', 100), hw2_manual.max_level(parsed_data2))
    print('Part1')
    assert_equals(('Lapras', 72), hw2_pandas.max_level(DATA))
    assert_equals(('Victreebel', 100), hw2_pandas.max_level(DATA2))
Example #3
0
def test_max_level():
    """
    Test the function max_level
    """
    print('Testing max_level')

    assert_equals(('Lapras', 72), hw2_manual.max_level(data1))
    assert_equals(('Lapras', 72), hw2_pandas.max_level(data2))
    assert_equals(('Victreebel', 100), hw2_manual.max_level(data3))
    assert_equals(('Victreebel', 100), hw2_pandas.max_level(data4))
    assert_equals(('Magmar', 96), hw2_manual.max_level(data5))
    assert_equals(('Magmar', 96), hw2_pandas.max_level(data6))
Example #4
0
def test_max_level():
    """
    Tests the function: max_level
    """
    print('Testing max_level')

    # Cases from the made up "spec" for this problem
    assert_equals(('Lapras', 72), hw2_manual.max_level(data1))
    assert_equals(('Lapras', 72), hw2_pandas.max_level(data2))
    # Additional two cases
    assert_equals(('Lapras', 72), hw2_manual.max_level(data3))
    assert_equals(('Lapras', 72), hw2_pandas.max_level(data4))
Example #5
0
def test_pandas_max_level():
    '''
    Test the pandas max_level function
    '''

    test_data = pd.read_csv(POKEMON_TEST_FILE)
    box_data = pd.read_csv(POKEMON_BOX_FILE)
    print('Testing max_level')

    # Spec test
    assert_equals(('Lapras', 72), hw2_pandas.max_level(test_data))

    # My test
    assert_equals(('Victreebel', 100), hw2_pandas.max_level(box_data))
def test_max_level():
    """
    Tests the output of both versions of max_level() function.
    """

    # Test Part 0 - Manual
    print('Test max_level from Part 0 - Manual')

    assert_equals(('Lapras', 72), hw2_manual.max_level(data_m1))
    assert_equals(('Victreebel', 100), hw2_manual.max_level(data_m2))

    # Test Part 1 - Pandas
    print('Test max_level from Part 1 - Pandas')

    assert_equals(('Lapras', 72), hw2_pandas.max_level(data_p1))
    assert_equals(('Victreebel', 100), hw2_pandas.max_level(data_p2))
Example #7
0
def part1():
    """
    Calculates various statistics about the Pokemon dataset using
    the implementation in Part 1.
    """
    print('=== Starting Part 1 ===')
    data = pd.read_csv(DATA)

    print('Number of species:', hw2_pandas.species_count(data))
    print('Highest level pokemon:', hw2_pandas.max_level(data))
    print('Low-level Pokemon', hw2_pandas.filter_range(data, 1, 9))
    print('Average attack for fire types',
          hw2_pandas.mean_attack_for_type(data, 'fire'))
    print('Count of each Pokemon type:')
    print(hw2_pandas.count_types(data))
    print('Highest stage for each Pokemon type')
    print(hw2_pandas.highest_stage_per_type(data))
    print('Average attack for each Pokemon type')
    print(hw2_pandas.mean_attack_per_type(data))