FilesExpand file tree

 preprint07

/

mem_test.py

Copy path

BlameMore file actions

BlameMore file actions

Latest commit

 

History

History

History

29 lines (23 loc) · 831 Bytes

 preprint07

/

mem_test.py

Top

File metadata and controls

• Code
• Blame

29 lines (23 loc) · 831 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

import importlib

import sys

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

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:,} {cls.__qualname__!r} instances')

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,}')

print(f' Final RAM usage: {mem_final:14,}')