import unittest

import astshim as ast
from astshim.test import MappingTestCase


class TestCmpFrame(MappingTestCase):

    def test_CmpFrame(self):
        frame1 = ast.Frame(2, "label(1)=a, label(2)=b")
        frame2 = ast.Frame(1, "label(1)=c")
        cmpframe = ast.CmpFrame(frame1, frame2)
        # adding to a CmpFrame increases by 1
        self.assertEqual(frame1.getRefCount(), 2)
        # adding to a CmpFrame increases by 1
        self.assertEqual(frame2.getRefCount(), 2)

        self.assertEqual(cmpframe.nAxes, 3)
        self.assertEqual(cmpframe.getLabel(1), "a")
        self.assertEqual(cmpframe.getLabel(2), "b")
        self.assertEqual(cmpframe.getLabel(3), "c")

        self.checkPersistence(cmpframe)
        self.checkMemoryForCompoundObject(frame1, frame2, cmpframe, isSeries=None)


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