import unittest

from .helpers import run_module


class MiscTests(unittest.TestCase):
    def setUp(self):
        self.name = "benchmarker"

    def test_no_framework(self):
        with self.assertRaises(Exception):
            run_module(self.name)

    def test_no_problem(self):
        with self.assertRaises(Exception):
            run_module(self.name, "--framework=pytorch")

Add tests to cover no framework, no problem