def test_getDataObject_givenData_expectGivenInputData(
self):
data = [1, 2, 3, 4]
dataCollection = DataCollection()
dataCollection.setDataObject(data, "test")
self.assertEqual(dataCollection.getDataObject("test"), data)
def test_getDataObject_givenNonExistingName_expectException(
self):
dataCollection = DataCollection()
self.assertRaises(
KeyError, lambda: dataCollection.getDataObject("noExisting"))
def test_getDataObject_givenEmptyName_expectException(
self):
dataCollection = DataCollection()
self.assertRaises(
KeyError, lambda: dataCollection.getDataObject(""))