from __future__ import absolute_import, print_function, division

import numpy as np

from .. import FITSFigure


def test_frame_linewidth():
    data = np.zeros((16, 16))
    f = FITSFigure(data)
    f.frame.set_linewidth(0)
    f.frame.set_linewidth(1)
    f.frame.set_linewidth(10)
    f.close()


def test_frame_color():
    data = np.zeros((16, 16))
    f = FITSFigure(data)
    f.frame.set_color('black')
    f.frame.set_color('#003344')
    f.frame.set_color((1.0, 0.4, 0.3))
    f.close()
