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'])