def test_Dataset_addFlatSFtoDataframe_exceptions(mockTree, mocker):
newDataset = Dataset("someName")
inputDF = copy.deepcopy(mockTree.dataframe)
with pytest.raises(RuntimeError):
newDataset.addFlatSFtoDataframe(inputDF)
with pytest.raises(TypeError):
newDataset.setSF("Hallo", "OkayName")
with pytest.raises(TypeError):
newDataset.setSF(1.2, [])
def test_Dataset_addFlatSFtoDataframe(mockTree, mocker):
SFName = "flatSF"
SF = 1.2
inputDF = copy.deepcopy(mockTree.dataframe)
expected = copy.deepcopy(mockTree.dataframe)
expected["flatSF"] = len(expected)*[SF]
print(expected)
newDataset = Dataset("someName")
newDataset.setSF(SF, SFName)
newDataset.addFlatSFtoDataframe(inputDF)
assert expected.equals(inputDF)