inheritance in tests

test.py

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vim:fenc=utf-8
#
# JSM product code
#
# (C) Copyright 2018 Juxt SmartMandate Pvt Ltd
# All right reserved.
#
# This file is confidential and NOT open source.  Do not distribute.
#

"""

"""

from unittest import TestCase, main


class TestBase(TestCase):

    @classmethod
    def setUpClass(cls):
        cls.some_attr = 'hello'

    def test_for_child(self):
        if not hasattr(self, 'CONST'):
            self.skipTest('Only for child.')
        self.assertTrue(isinstance(self.CONST, int))

    def test_for_parent_and_child(self):
        self.assertTrue(True)


class MyClass(TestBase):

    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        cls.CONST = 123

    def test_exclusive_to_child(self):
        self.assertTrue(True)


if __name__ == '__main__':
    main()