def testAppendAndFlushBufferSize(self): """ Test that the append function flushes when the buffer size is large. """ # Setup data. simulation_time = 12.3 step = 14 class Config: def __init__(self): pass configuration = Config() # Instantiate a trajectory. t = Trajectory(trajectory_filename="filename") # Add the needed functions. def storeData(*args, **kwargs): pass t._storeData = storeData ret_buffer = 10.0 def bufferSize(*args, **kwargs): return ret_buffer t._bufferSize = bufferSize class Dummy: self.fc = False d = Dummy() def flush(): d.fc = True t.flush = flush # Append. t.append(simulation_time, step, configuration) # The first time allways flushes. self.assertTrue(d.fc) # Append again. d.fc = False t.append(simulation_time, step, configuration) self.assertFalse(d.fc) # Change the value of the buffer size so that flush() gets called. ret_buffer = t._Trajectory__max_buffer_size + 1 # Append. t.append(simulation_time, step, configuration) # Check. self.assertTrue(d.fc)
def testAppendInterface(self): """ Test that the append function can not be called directly from the baseclass. """ # Setup data. simulation_time = 12.3 step = 14 class Config: def __init__(self): pass configuration = Config() # Instantiate a trajectory. t = Trajectory(trajectory_filename="filename") # Call append. self.assertRaises( AttributeError, lambda: t.append(simulation_time, step, configuration)) # Add the missing function. def storeData(*args, **kwargs): pass t._storeData = storeData self.assertRaises( AttributeError, lambda: t.append(simulation_time, step, configuration)) # Add the missing function. def bufferSize(*args, **kwargs): pass t._bufferSize = bufferSize self.assertRaises( AttributeError, lambda: t.append(simulation_time, step, configuration)) # Add the missing function. def flush(*args, **kwargs): pass t.flush = flush # Now it works. t.append(simulation_time, step, configuration)
def testAppendInterface(self): """ Test that the append function can not be called directly from the baseclass. """ # Setup data. simulation_time = 12.3 step = 14 class Config: def __init__(self): pass configuration = Config() # Instantiate a trajectory. t = Trajectory(trajectory_filename="filename") # Call append. self.assertRaises(AttributeError, lambda: t.append(simulation_time, step, configuration)) # Add the missing function. def storeData(*args, **kwargs): pass t._storeData = storeData self.assertRaises(AttributeError, lambda: t.append(simulation_time, step, configuration)) # Add the missing function. def bufferSize(*args, **kwargs): pass t._bufferSize = bufferSize self.assertRaises(AttributeError, lambda: t.append(simulation_time, step, configuration)) # Add the missing function. def flush(*args, **kwargs): pass t.flush = flush # Now it works. t.append(simulation_time, step, configuration)
def testAppendTimeUpdate(self): """ Test that the append function updates the time only when it flushes. """ # Setup data. simulation_time = 12.3 step = 14 class Config: def __init__(self): pass configuration = Config() # Instantiate a trajectory. t = Trajectory(trajectory_filename="filename") # Add the needed functions. def storeData(*args, **kwargs): pass t._storeData = storeData ret_buffer = 10.0 def bufferSize(*args, **kwargs): return ret_buffer t._bufferSize = bufferSize class Dummy: self.fc = False d = Dummy() def flush(): d.fc = True t.flush = flush # Append. t.append(simulation_time, step, configuration) # The first time allways flushes. self.assertTrue(d.fc) # This updates the time last dump variable. ref_time_last_dump = t._Trajectory__time_last_dump # Sleep 10 ms. time.sleep(0.01) # Append again. d.fc = False t.append(simulation_time, step, configuration) self.assertFalse(d.fc) # This does not update the time last dump. self.assertAlmostEqual(ref_time_last_dump, t._Trajectory__time_last_dump, 10) # Change the value of the max buffer time so that flush() gets called. t._Trajectory__max_buffer_time = 0.0 # Sleep 10 ms. time.sleep(0.01) # Append. t.append(simulation_time, step, configuration) # Check. self.assertTrue(d.fc) # The time last dump should have been updated. self.assertTrue( (t._Trajectory__time_last_dump - ref_time_last_dump) > 0.0001 )
def testAppendTimeUpdate(self): """ Test that the append function updates the time only when it flushes. """ # Setup data. simulation_time = 12.3 step = 14 class Config: def __init__(self): pass configuration = Config() # Instantiate a trajectory. t = Trajectory(trajectory_filename="filename") # Add the needed functions. def storeData(*args, **kwargs): pass t._storeData = storeData ret_buffer = 10.0 def bufferSize(*args, **kwargs): return ret_buffer t._bufferSize = bufferSize class Dummy: self.fc = False d = Dummy() def flush(): d.fc = True t.flush = flush # Append. t.append(simulation_time, step, configuration) # The first time allways flushes. self.assertTrue(d.fc) # This updates the time last dump variable. ref_time_last_dump = t._Trajectory__time_last_dump # Sleep 10 ms. time.sleep(0.01) # Append again. d.fc = False t.append(simulation_time, step, configuration) self.assertFalse(d.fc) # This does not update the time last dump. self.assertAlmostEqual(ref_time_last_dump, t._Trajectory__time_last_dump, 10) # Change the value of the max buffer time so that flush() gets called. t._Trajectory__max_buffer_time = 0.0 # Sleep 10 ms. time.sleep(0.01) # Append. t.append(simulation_time, step, configuration) # Check. self.assertTrue(d.fc) # The time last dump should have been updated. self.assertTrue( (t._Trajectory__time_last_dump - ref_time_last_dump) > 0.0001)