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)
