38 lines
960 B
Python
38 lines
960 B
Python
|
import matplotlib.gridspec as gridspec
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def test_equal():
|
||
|
|
gs = gridspec.GridSpec(2, 1)
|
||
|
|
assert gs[0, 0] == gs[0, 0]
|
||
|
|
assert gs[:, 0] == gs[:, 0]
|
||
|
|
|
||
|
|
|
||
|
|
def test_width_ratios():
|
||
|
|
"""
|
||
|
|
Addresses issue #5835.
|
||
|
|
See at https://github.com/matplotlib/matplotlib/issues/5835.
|
||
|
|
"""
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
gridspec.GridSpec(1, 1, width_ratios=[2, 1, 3])
|
||
|
|
|
||
|
|
|
||
|
|
def test_height_ratios():
|
||
|
|
"""
|
||
|
|
Addresses issue #5835.
|
||
|
|
See at https://github.com/matplotlib/matplotlib/issues/5835.
|
||
|
|
"""
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
gridspec.GridSpec(1, 1, height_ratios=[2, 1, 3])
|
||
|
|
|
||
|
|
|
||
|
|
def test_repr():
|
||
|
|
ss = gridspec.GridSpec(3, 3)[2, 1:3]
|
||
|
|
assert repr(ss) == "GridSpec(3, 3)[2:3, 1:3]"
|
||
|
|
|
||
|
|
ss = gridspec.GridSpec(2, 2,
|
||
|
|
height_ratios=(3, 1),
|
||
|
|
width_ratios=(1, 3))
|
||
|
|
assert repr(ss) == \
|
||
|
|
"GridSpec(2, 2, height_ratios=(3, 1), width_ratios=(1, 3))"
|