def test_push_modules_and_classes(self):
modules = ['sample', 'a_name']
test_classes = ['sampleTests', 'a_class_name']
gen = CompileTags(source_module_name='sample')
gen.push_modules_and_classes(modules=['a_name'], test_classes=test_classes, context=self)
self.assertEqual(gen.unit_test_module, modules)
self.assertEqual(gen.unit_test_class, test_classes)
def test_pop_module_and_class(self):
modules = ['sample', 'a_name']
test_classes = ['sampleTests', 'a_class_name']
gen = CompileTags(source_module_name='sample.py')
# --> First push
gen.push_modules_and_classes(modules=modules, test_classes=test_classes, context=self)
self.assertEqual(gen.unit_test_module, modules)
self.assertEqual(gen.unit_test_class, test_classes)
# --> Then test pop
gen.pop_module_and_class(modules=modules, test_classes=test_classes)
self.assertEqual(len(gen.unit_test_module), 1)
self.assertEqual(len(gen.unit_test_class), 1)
