import numpy as np
import hexy as hx

def test_get_hex_line():
    expected = [
            [-3, 3, 0],
            [-2, 2, 0],
            [-1, 2, -1],
            [0, 2, -2],
            [1, 1, -2],
            ]
    start = np.array([-3, 3, 0])
    end = np.array([1, 1, -2])
    print(hx.get_hex_line(start, end))
    print(expected);
    assert(np.array_equal(
        hx.get_hex_line(start, end),
        expected));

if __name__ == "__main__":
    test_get_hex_line()

[((12, 12, 12, 32), 'numpy.array', 'np.array', ({(12, 21, 12, 31): '[-3, 3, 0]'}, {}), '([-3, 3, 0])', True, 'import numpy as np\n'), ((13, 10, 13, 30), 'numpy.array', 'np.array', ({(13, 19, 13, 29): '[1, 1, -2]'}, {}), '([1, 1, -2])', True, 'import numpy as np\n'), ((14, 10, 14, 37), 'hexy.get_hex_line', 'hx.get_hex_line', ({(14, 26, 14, 31): 'start', (14, 33, 14, 36): 'end'}, {}), '(start, end)', True, 'import hexy as hx\n'), ((17, 8, 17, 35), 'hexy.get_hex_line', 'hx.get_hex_line', ({(17, 24, 17, 29): 'start', (17, 31, 17, 34): 'end'}, {}), '(start, end)', True, 'import hexy as hx\n')]