xref: /external/trappy/tests/test_dynamic.py

# $Copyright:
# ----------------------------------------------------------------
# This confidential and proprietary software may be used only as
# authorised by a licensing agreement from ARM Limited
#  (C) COPYRIGHT 2015 ARM Limited
#       ALL RIGHTS RESERVED
# The entire notice above must be reproduced on all authorised
# copies and copies may only be made to the extent permitted
# by a licensing agreement from ARM Limited.
# ----------------------------------------------------------------
# File:        tests_dynamic.py
# ----------------------------------------------------------------
# $
#

import unittest
import matplotlib
from test_sched import BaseTestSched
from cr2.base import Base
import cr2


class DynamicEvent(Base):

    """Test the ability to register
       specific classes to cr2"""

    unique_word = "dynamic_test_key"
    name = "dynamic_event"

    def __init__(self, path=None):
        super(DynamicEvent, self).__init__(
            basepath=path,
            unique_word=self.unique_word,
        )


class TestDynamicEvents(BaseTestSched):

    def __init__(self, *args, **kwargs):
        super(TestDynamicEvents, self).__init__(*args, **kwargs)

    def test_dynamic_data_frame(self):
        """
           Test if the dynamic events are populated
           in the data frame
        """
        cr2.register_dynamic("DynamicEvent", "dynamic_test_key")
        r = cr2.Run(name="first")
        self.assertTrue(len(r.dynamic_event.data_frame) == 1)

    def test_dynamic_class_attr(self):
        """
           Test the attibutes of the dynamically
           generated class
        """
        cls = cr2.register_dynamic("DynamicEvent", "dynamic_test_key")
        self.assertEquals(cls.__name__, "DynamicEvent")
        self.assertEquals(cls.name, "dynamic_event")
        self.assertEquals(cls.unique_word, "dynamic_test_key")

    def test_dynamic_event_plot(self):
        """Test if plotter can accept a dynamic class
            for a template argument"""

        cls = cr2.register_dynamic("DynamicEvent", "dynamic_test_key")
        r = cr2.Run(name="first")
        l = cr2.LinePlot(r, cls, column="load")
        l.view()
        matplotlib.pyplot.close('all')

    def test_dynamic_event_scope(self):
	"""Test the case when an "all" scope class is
	registered. it should appear in both thermal and sched
	run class definitions when scoped run objects are created
	"""
        cls = cr2.register_dynamic("DynamicEvent", "dynamic_test_key")
        r1 = cr2.Run(name="first")
	print r1.class_definitions
	self.assertTrue(r1.class_definitions.has_key(cls.name))

    def test_register_class(self):
        cr2.register_class(DynamicEvent)
        r = cr2.Run(name="first")
        self.assertTrue(len(r.dynamic_event.data_frame) == 1)