Wie pytest Platz Speichergrenzen für Tests?

Question

Ich benutze Pytest, aber ich hätte gerne einen Dekorator, der eine maximale Speicherauslastung pro Test einstellen könnte. Ähnlich wie bei diesem question die mit beantwortet wurde,

@pytest.mark.timeout(300) 
def test_foo(): 
    pass 

Ich möchte,

@pytest.mark.maxmem(300) 
def test_foo(): 
    pass 

EDIT:

Ich habe versucht,

>>> import os, psutil 
>>> import numpy as np 
>>> process = psutil.Process(os.getpid()) 
>>> process.memory_info().rss/1e9 
0.01978368 
>>> def f(): 
...  x = np.arange(int(1e9)) 
... 
>>> process.memory_info().rss/1e9 
0.01982464 
>>> f() 
>>> process.memory_info().rss/1e9 
0.019832832 

Welche fangen nicht die Speicherzuweisung in der Funktion.

Answer 1

Nachdem ich gelernt habe, wie limit the memory used und zu sehen, wie viel memory is currently used, schrieb ich einen Dekorator, der aus, wenn das Speicherinkrement zu hoch ist. Es ist ein bisschen buggy mit der Festlegung der Grenzen, aber es funktioniert gut genug für mich.

import resource, os, psutil 
import numpy 

def memory_limit(max_mem): 
    def decorator(f): 
     def wrapper(*args, **kwargs): 
      process = psutil.Process(os.getpid()) 
      prev_limits = resource.getrlimit(resource.RLIMIT_AS) 
      resource.setrlimit(resource.RLIMIT_AS, (process.memory_info().rss + max_mem, -1)) 
      result = f(*args, **kwargs) 
      resource.setrlimit(resource.RLIMIT_AS, prev_limits) 
      return result 
     return wrapper 
    return decorator 


@memory_limit(int(16e8)) 
def allocate(N): 
    return numpy.arange(N, dtype='u8') 

a = [allocate(int(1e8)) for i in range(10)] 

try: 
    allocate(int(3e8)) 
except: 
    exit(0) 
raise Exception("Should have failed") 

Zumindest läuft auf meinem Computer Code und endet ohne einen Fehler.