mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 20:42:26 +00:00
81337b4fdd
* Reduced test time by 85% using vcrpy * Refactored tests with decorator * Added vcrpy dependencies * Fixed issues with tests * Fixed tests not running in local Co-authored-by: Colin Delahunty <colindelahunty@Colins-Air.attlocal.net>
27 lines
719 B
Python
27 lines
719 B
Python
import functools
|
|
import sys
|
|
import io
|
|
|
|
|
|
def check_print(assert_in: str = "", length: int = -1):
|
|
"""Captures output of print function and checks if the function contains a given string"""
|
|
|
|
def checker(func):
|
|
@functools.wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
capturedOutput = io.StringIO()
|
|
sys.stdout = capturedOutput
|
|
func(*args, **kwargs)
|
|
sys.stdout = sys.__stdout__
|
|
capt = capturedOutput.getvalue()
|
|
if assert_in:
|
|
assert assert_in in capt
|
|
return None
|
|
if length >= 0:
|
|
assert len(capt) > length
|
|
return capt
|
|
|
|
return wrapper
|
|
|
|
return checker
|