class TestSimpleBC: df = bcr.load_dataset('covid19') def test_defaults(self): bcr.bar_chart_race(self.df.iloc[-20:-16]) bcr.bar_chart_race(self.df.iloc[-20:-16], orientation='v') def test_sort(self): bcr.bar_chart_race(self.df.iloc[-20:-16], sort='asc') bcr.bar_chart_race(self.df.iloc[-20:-16], orientation='v', sort='asc') def test_nbars(self): bcr.bar_chart_race(self.df.iloc[-20:-16], sort='desc', n_bars=8) bcr.bar_chart_race(self.df.iloc[-20:-16], orientation='v', sort='desc', n_bars=8) def fixed_order(self): bcr.bar_chart_race(df.iloc[-20:-16], sort='asc', n_bars=8, fixed_order=True) bcr.bar_chart_race(df.iloc[-20:-16], sort='asc', fixed_order=['Iran', 'USA', 'Italy', 'Spain'])
class TestSimpleBC: df = bcr.load_dataset('covid19') df = df.iloc[-20:-16] df1 = df.reset_index(drop=True) def test_defaults(self): bcr.bar_chart_race(self.df) bcr.bar_chart_race(self.df, orientation='v') def test_sort(self): bcr.bar_chart_race(self.df, sort='asc') bcr.bar_chart_race(self.df, orientation='v', sort='asc') def test_nbars(self): bcr.bar_chart_race(self.df, sort='desc', n_bars=8) bcr.bar_chart_race(self.df, orientation='v', sort='desc', n_bars=8) def test_fixed_order(self): bcr.bar_chart_race(self.df), sort='asc', n_bars=8, fixed_order=True) bcr.bar_chart_race(self.df, fixed_order=['Iran', 'USA', 'Italy', 'Spain']) def test_fixed_max(self): bcr.bar_chart_race(self.df, fixed_max=True) def test_steps_per_period(self): bcr.bar_chart_race(self.df, sort='asc', steps_per_period=2) bcr.bar_chart_race(self.df, sort='asc', steps_per_period=30) def test_interpolate_period(self): bcr.bar_chart_race(self.df, interpolate_period=True, figsize=(5, 3), n_bars=8) def test_label_bars(self): bcr.bar_chart_race(self.df, figsize=(5, 3), n_bars=8, label_bars=False) def test_bar_size(self): bcr.bar_chart_race(self.df, figsize=(4, 2.5), n_bars=8, bar_size=.99) def test_period_label(self): bcr.bar_chart_race(self.df, figsize=(4, 2.5), n_bars=8, period_label=False) bcr.bar_chart_race(self.df, figsize=(4, 2.5), n_bars=8, period_label={'x': .99, 'y': .1, 'ha': 'right'})
import bar_chart_race as bcr df = bcr.load_dataset('covid19_tutorial') bcr.bar_chart_race(df=df, filename='./people_movement.gif', orientation='h', sort='desc', n_bars=8, fixed_order=False, fixed_max=True, steps_per_period=20, period_length=500, end_period_pause=0, interpolate_period=False, period_label={ 'x': .98, 'y': .3, 'ha': 'right', 'va': 'center' }, period_template='%B %d, %Y', period_summary_func=lambda v, r: { 'x': .98, 'y': .2, 's': f'Total deaths: {v.sum():,.0f}', 'ha': 'right', 'size': 11 }, perpendicular_bar_func='median', colors='dark12', title='People Movement', bar_size=.95,
def test_load_covid(self): bcr.load_dataset('covid19')
def test_load_urban_pop(self): bcr.load_dataset('urban_pop')
import numpy as np import pandas as pd import matplotlib.pyplot as plt import bar_chart_race as bcr import plotly.graph_objects as go import plotly from plotly.subplots import make_subplots from bar_chart_race import load_dataset, bar_chart_race_plotly df = load_dataset('covid19') df = df.iloc[-20:-10] df1 = df.reset_index(drop=True) def test_all(): bar_chart_race_plotly(df) df.bcr.bar_chart_race_plotly() bar_chart_race_plotly(df, sort='asc') bar_chart_race_plotly(df, orientation='v', slider=False) bar_chart_race_plotly(df, orientation='v', sort='asc', slider=False) bar_chart_race_plotly(df, sort='desc', n_bars=8) bar_chart_race_plotly(df, orientation='v', sort='desc', n_bars=8) bar_chart_race_plotly(df, sort='asc', n_bars=8, fixed_order=True)
import pytest import bar_chart_race._utils as utils from bar_chart_race import load_dataset, bar_chart_race df = load_dataset('baseball') df = df.iloc[-20:-16] def test_threshold(): filtered_df = utils.filter_threshold(df, 60) assert len(filtered_df) == 1 filtered_df = utils.filter_threshold(df, 0) assert len(filtered_df) == 4 filtered_df = utils.filter_threshold(df, 50) assert filtered_df.iloc[0]['hr'] > 50