import unittest

from site_up import update_lamp_status


class TestSiteUp(unittest.TestCase):

    def test_lamp_is_set_to_red_when_site_is_down(self):
        # TODO
        pass

    def test_lamp_is_set_to_green_when_site_is_up(self):
        # TODO
        pass


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

import unittest

from site_up import update_lamp_status


class TestSiteUp(unittest.TestCase):

    def test_lamp_is_set_to_red_when_site_is_down(self):
        # TODO
        pass

    def test_lamp_is_set_to_green_when_site_is_up(self):
        # TODO
        pass


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

# TODO # TODO