def test_create_class(self): class_metadata = {'module': 'collections', 'name': 'Counter'} cls = BW2Package._create_class(class_metadata, False) import collections self.assertEqual(cls, collections.Counter) class_metadata = {'module': 'bw2data.method', 'name': 'Method'} cls = BW2Package._create_class(class_metadata, False) self.assertEqual(cls, Method)
def test_create_class(self): class_metadata = {"module": "collections", "name": "Counter"} cls = BW2Package._create_class(class_metadata, False) import collections self.assertEqual(cls, collections.Counter) class_metadata = {"module": "bw2data.method", "name": "Method"} cls = BW2Package._create_class(class_metadata, False) self.assertEqual(cls, Method)
def test_create_class_whitelist(self): bad_class_metadata = { 'module': 'some.package', 'name': 'Foo', } with self.assertRaises(UnsafeData): BW2Package._create_class(bad_class_metadata) with self.assertRaises(ImportError): BW2Package._create_class(bad_class_metadata, False)
def test_create_class_whitelist(self): bad_class_metadata = { "module": "some.package", "name": "Foo", } with self.assertRaises(UnsafeData): BW2Package._create_class(bad_class_metadata) with self.assertRaises(ImportError): BW2Package._create_class(bad_class_metadata, False)