cvmodelz/tests/model_tests/factory_tests.py

import unittest

from cvmodelz.models import ModelFactory


class FactoryTests(unittest.TestCase):

	def test_model_creation(self):
		for key in ModelFactory.get_all_models():
			msg = f"Creating {key}..."
			print(msg)
			model = ModelFactory.new(key)
			print("\033[A{}\033[A".format(" "*len(msg)))
			self.assertIsNotNone(model)