181 lines
3.8 KiB
Python
181 lines
3.8 KiB
Python
import nafuma.plotting as btp
|
|
from cycler import cycler
|
|
import itertools
|
|
import numpy as np
|
|
|
|
import matplotlib.pyplot as plt
|
|
import matplotlib as mpl
|
|
|
|
|
|
def test_generate_colours() -> None:
|
|
|
|
assert type(btp.generate_colours('black', kind='single')) == itertools.cycle
|
|
|
|
palettes = [('qualitative', 'Dark2_8')]
|
|
colour_cycle = btp.generate_colours(palettes)
|
|
|
|
assert type(colour_cycle) == itertools.cycle
|
|
|
|
|
|
# Test that it actually loaded 8 colours when given a set of 8 colours to
|
|
|
|
same_colour = None
|
|
for i in range(10):
|
|
colour = next(colour_cycle)
|
|
if i == 0:
|
|
first_colour = colour
|
|
|
|
if colour == first_colour:
|
|
repeat_colour_index = i
|
|
|
|
|
|
assert repeat_colour_index == 8
|
|
|
|
|
|
|
|
|
|
def test_update_rc_params() -> None:
|
|
|
|
rc_params = {
|
|
'lines.linewidth': 100
|
|
}
|
|
|
|
prev_params = plt.rcParams['lines.linewidth']
|
|
|
|
# Update run commands if any is passed (will pass an empty dictionary if not passed)
|
|
btp.update_rc_params(rc_params)
|
|
|
|
new_params = plt.rcParams['lines.linewidth']
|
|
|
|
assert new_params == 100
|
|
assert prev_params != new_params
|
|
|
|
|
|
# Reset run commands
|
|
plt.rcdefaults()
|
|
|
|
|
|
|
|
def test_scale_figure() -> None:
|
|
|
|
width, height = 1, 1
|
|
|
|
format_params = {
|
|
'upscaling_factor': 2,
|
|
'compress_width': 1,
|
|
'compress_height': 1
|
|
}
|
|
|
|
width1, height1 = btp.scale_figure(format_params=format_params, width=width, height=height)
|
|
|
|
assert width1 == 2 and height1 == 2
|
|
|
|
format_params = {
|
|
'upscaling_factor': 1,
|
|
'compress_width': 0.5,
|
|
'compress_height': 1
|
|
}
|
|
|
|
width2, height2 = btp.scale_figure(format_params=format_params, width=width, height=height)
|
|
|
|
assert width2 == 0.5 and height2 == 1
|
|
|
|
format_params = {
|
|
'upscaling_factor': 2,
|
|
'compress_width': 0.5,
|
|
'compress_height': 0.2
|
|
}
|
|
|
|
width2, height2 = btp.scale_figure(format_params=format_params, width=width, height=height)
|
|
|
|
assert width2 == 1 and height2 == 0.4
|
|
|
|
|
|
def test_determine_width() -> None:
|
|
|
|
conversion_cm_inch = 0.3937008 # cm to inch
|
|
|
|
format_params = {
|
|
'column_type': 'single',
|
|
'single_column_width': 5,
|
|
'double_column_width': 10,
|
|
'width_ratio': '1:1'
|
|
}
|
|
|
|
assert np.round(btp.determine_width(format_params),6) == np.round(5*conversion_cm_inch,6)
|
|
|
|
format_params['column_type'] = 'double'
|
|
|
|
assert np.round(btp.determine_width(format_params), 6) == np.round(10*conversion_cm_inch, 6)
|
|
|
|
|
|
format_params['column_type'] = 'single'
|
|
format_params['width_ratio'] = '1:2'
|
|
|
|
assert np.round(btp.determine_width(format_params), 6) == np.round(2.5*conversion_cm_inch, 6)
|
|
|
|
def test_determine_height() -> None:
|
|
|
|
|
|
width = 1
|
|
|
|
format_params = {
|
|
'aspect_ratio': '1:1'
|
|
}
|
|
|
|
assert btp.determine_height(format_params=format_params, width=width) == 1
|
|
|
|
format_params['aspect_ratio'] = '3:1'
|
|
|
|
assert (btp.determine_height(format_params=format_params, width=width) - 0.333333333333333) < 10e-7
|
|
|
|
assert True
|
|
|
|
|
|
|
|
def test_prepare_plot() -> None:
|
|
|
|
fig, ax = btp.prepare_plot()
|
|
|
|
assert type(fig) == plt.Figure
|
|
assert fig.get_dpi() == 600
|
|
assert ax.get_xlim() == (0.0, 1.0)
|
|
|
|
|
|
|
|
def test_adjust_plot() -> None:
|
|
|
|
fig, ax = btp.prepare_plot()
|
|
|
|
options = {
|
|
'xlim': (0.0, 2.0),
|
|
'title': 'Test'
|
|
}
|
|
|
|
fig, ax = btp.adjust_plot(fig, ax, options)
|
|
|
|
|
|
assert ax.get_xlim() == (0.0, 2.0)
|
|
assert ax.get_title() == 'Test'
|
|
|
|
|
|
|
|
def test_ipywidgets_update() -> None:
|
|
|
|
|
|
|
|
def test_func(data, options):
|
|
test1 = options['test1']
|
|
test2 = options['test2']
|
|
|
|
assert type(data) == dict
|
|
assert test1 == 1
|
|
assert test2 == 2
|
|
|
|
|
|
|
|
data = {}
|
|
options = {}
|
|
|
|
btp.ipywidgets_update(func=test_func, data=data, options=options, test1=1, test2=2)
|
|
|