def test_setOutputVariables_givenSingleOutputVariable_expectSetOutputVariables( self): dataManager = DataManager('values') dataManager.addDataEntry('X', 1) dataManager.addDataEntry('Y', 1) dataManager.addDataEntry('A', 1) mapping = Mapping(dataManager, ['X'], ['Y'], "TestMapping") mapping.setOutputVariables(['A']) self.assertEqual(mapping.getOutputVariables(), ['A'])
def test_setInputVariables_givenInputVariablesAppend_expectConstructorAndSetInputVariables( self): dataManager = DataManager('values') dataManager.addDataEntry('X', 1) dataManager.addDataEntry('Y', 1) dataManager.addDataEntry('A', 1) dataManager.addDataEntry('B', 1) mapping = Mapping(dataManager, ['X'], ['Y'], "TestMapping") mapping.setInputVariables(['A', 'B'], None, True) self.assertEqual(mapping.getInputVariables(), ['X', 'A', 'B'])
def test_setInputVariables_givenNumDim_expectNotSupportet(self): ''' This test may fail if the functionality gets implemented ''' dataManager = DataManager('values') dataManager.addDataEntry('X', 1) dataManager.addDataEntry('Y', 1) dataManager.addDataEntry('A', 1) dataManager.addDataEntry('B', 1) mapping = Mapping(dataManager, ['X'], ['Y'], "TestMapping") self.assertRaises(NotImplementedError, lambda: mapping.setInputVariables(['A', 'B'], 5))
def test_setInputVariables_givenNumberInInputvariables_expectNotSupportet( self): ''' This test may fail if the functionality gets implemented ''' dataManager = DataManager('values') dataManager.addDataEntry('X', 1) dataManager.addDataEntry('Y', 1) dataManager.addDataEntry('A', 1) dataManager.addDataEntry('B', 1) mapping = Mapping(dataManager, ['X'], ['Y'], "TestMapping") self.assertRaises( DeprecationWarning, lambda: mapping.setInputVariables([5, 'A', 'B'], None, True))
def __init__(self, dataManager): Mapping.__init__(self, dataManager, ['states', 'actions'], ['rewards'])
def __init__(self, dataManager, inputVariables=['states', 'actions'], outputVariables=['nextStates']): Mapping.__init__(self, dataManager, inputVariables, outputVariables)
def __init__(self, dataManager): Mapping.__init__(self, dataManager, inputVariables=['X'], outputVariables='X')
def __init__(self, dataManager): Mapping.__init__(self, dataManager, ['X'], ['Y'])