updated from Atlas
This commit is contained in:
@@ -4,20 +4,25 @@ import resource
|
||||
|
||||
NUM_VECTORS = 10**7
|
||||
|
||||
module = None
|
||||
if len(sys.argv) == 2:
|
||||
module_name = sys.argv[1].replace('.py', '')
|
||||
module = importlib.import_module(module_name)
|
||||
else:
|
||||
print(f'Usage: {sys.argv[0]} <vector-module-to-test>')
|
||||
sys.exit(2) # command line usage error
|
||||
|
||||
fmt = 'Selected Vector2d type: {.__name__}.{.__name__}'
|
||||
print(fmt.format(module, module.Vector2d))
|
||||
if module is None:
|
||||
print('Running test with built-in `complex`')
|
||||
cls = complex
|
||||
else:
|
||||
fmt = 'Selected Vector2d type: {.__name__}.{.__name__}'
|
||||
print(fmt.format(module, module.Vector2d))
|
||||
cls = module.Vector2d
|
||||
|
||||
mem_init = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
|
||||
print(f'Creating {NUM_VECTORS:,} Vector2d instances')
|
||||
print(f'Creating {NUM_VECTORS:,} {cls.__qualname__!r} instances')
|
||||
|
||||
vectors = [module.Vector2d(3.0, 4.0) for i in range(NUM_VECTORS)]
|
||||
vectors = [cls(3.0, 4.0) for i in range(NUM_VECTORS)]
|
||||
|
||||
mem_final = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
|
||||
print(f'Initial RAM usage: {mem_init:14,}')
|
||||
|
||||
Reference in New Issue
Block a user