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
