import unittest

import numpy as np

from damaris.data.Error_Result import Error_Result


class TestErrorResult(unittest.TestCase):
    def test_plotdata_are_ndarrays(self):
        """get_xdata/get_ydata must return numpy arrays like every Drawable."""
        er = Error_Result("core failed")
        self.assertIsInstance(er.get_xdata(), np.ndarray)
        self.assertIsInstance(er.get_ydata(), np.ndarray)
        np.testing.assert_array_equal(er.get_xdata(), [0.0])
        np.testing.assert_array_equal(er.get_ydata(), [0.0])

    def test_add_returns_self(self):
        er = Error_Result("msg")
        self.assertIs(er + 1, er)


if __name__ == "__main__":
    unittest.main()