# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE

from __future__ import annotations

import numpy as np
import pytest

import awkward as ak

behavior = {}


@ak.behaviors.mixins.mixin_class(behavior, "1030_point")
class AVeryVeryVeryLargeClassName:
    @property
    def mag2(self):
        return self.x**2 + self.y**2


def test():
    x = np.random.randint(0, 64, 128)
    y = np.random.randint(0, 64, 128)

    point_1030 = ak.operations.zip(
        {"x": x, "y": y}, behavior=behavior, with_name="1030_point"
    )
    assert isinstance(point_1030.mag2, ak.Array)

    point = ak.operations.zip(
        {"x": x, "y": y}, behavior=behavior, with_name="AVeryVeryVeryLargeClassName"
    )
    with pytest.raises(AttributeError):
        assert isinstance(point.mag2, ak.Array)