Skip to content

Commit

Permalink
remove vis used only with crim data
Browse files Browse the repository at this point in the history
  • Loading branch information
RichardFreedman committed Nov 17, 2023
1 parent 4bcc36c commit fc02773
Showing 1 changed file with 42 additions and 42 deletions.
84 changes: 42 additions & 42 deletions test/test_vis.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,45 +179,45 @@ def test_plot_close_match_heatmap():
# helper_test_generate_ngrams_and_dur(model, mel_chromatic, 5)
# # helper_test_generate_ngrams_and_dur(model, mel_chromatic, -1)


def test_comparisons_heatmap():
# pieces
df_relationships = pd.DataFrame(RELATIONSHIPS_DICT_EXAMPLE)
relationships_chart = viz.plot_comparison_heatmap(df_relationships, 'model_observation.ema',
main_category='relationship_type', other_category='observer.name',
heat_map_width=800, heat_map_height=300)
assert isinstance(relationships_chart, alt.VConcatChart)
assert len(relationships_chart.vconcat) == 2

df_observations = pd.DataFrame(OBSERVATIONS_DICT_EXAMPLE)
observations_chart = viz.plot_comparison_heatmap(df_observations, 'ema',
main_category='musical_type', other_category='observer.name',
heat_map_width=800, heat_map_height=300)
assert isinstance(observations_chart, alt.VConcatChart)
assert len(observations_chart.vconcat) == 2


def test_generate_networks_and_interactive_df():
df_observations = pd.DataFrame(OBSERVATIONS_DICT_EXAMPLE)

# time, pe
pen_networks, pen_widget = viz.create_comparisons_networks_and_interactive_df(df_observations, 'mt_pe_tint', 'time',
'ema')
assert pen_networks
assert pen_widget

# melodic, fug
fug_networks, fug_widget = viz.create_comparisons_networks_and_interactive_df(df_observations, 'mt_fg_int',
'melodic',
'ema')
assert fug_networks
assert fug_widget

patterns = df_observations['mt_fg_int'].to_list()
fug_networks_filtered, fug_widget_filtered = viz.create_comparisons_networks_and_interactive_df(df_observations,
'mt_fg_int',
'melodic', 'ema',
patterns)

assert fug_networks_filtered
assert fug_widget_filtered
# the following are only for vis maps using CRIM Django data. Not needed for Streamlit
# def test_comparisons_heatmap():
# # pieces
# df_relationships = pd.DataFrame(RELATIONSHIPS_DICT_EXAMPLE)
# relationships_chart = viz.plot_comparison_heatmap(df_relationships, 'model_observation.ema',
# main_category='relationship_type', other_category='observer.name',
# heat_map_width=800, heat_map_height=300)
# assert isinstance(relationships_chart, alt.VConcatChart)
# assert len(relationships_chart.vconcat) == 2

# df_observations = pd.DataFrame(OBSERVATIONS_DICT_EXAMPLE)
# observations_chart = viz.plot_comparison_heatmap(df_observations, 'ema',
# main_category='musical_type', other_category='observer.name',
# heat_map_width=800, heat_map_height=300)
# assert isinstance(observations_chart, alt.VConcatChart)
# assert len(observations_chart.vconcat) == 2


# def test_generate_networks_and_interactive_df():
# df_observations = pd.DataFrame(OBSERVATIONS_DICT_EXAMPLE)

# # time, pe
# pen_networks, pen_widget = viz.create_comparisons_networks_and_interactive_df(df_observations, 'mt_pe_tint', 'time',
# 'ema')
# assert pen_networks
# assert pen_widget

# # melodic, fug
# fug_networks, fug_widget = viz.create_comparisons_networks_and_interactive_df(df_observations, 'mt_fg_int',
# 'melodic',
# 'ema')
# assert fug_networks
# assert fug_widget

# patterns = df_observations['mt_fg_int'].to_list()
# fug_networks_filtered, fug_widget_filtered = viz.create_comparisons_networks_and_interactive_df(df_observations,
# 'mt_fg_int',
# 'melodic', 'ema',
# patterns)

# assert fug_networks_filtered
# assert fug_widget_filtered

0 comments on commit fc02773

Please sign in to comment.