mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 04:31:13 +00:00
cca3b31856
* updating dependencies * poetry export * removed breakline from input * Export dependencies * Workflow * Tests * Workflow * Dependencies update * Linting * Tests readme * Dependencies update * Workflow update * Linting * Workflow update * Fix import * Tests * Linting * Workflow update * Re-solve the dependencies with poetry 1.1.13 * Workflow update Co-authored-by: James Maslek <jmaslek11@gmail.com> Co-authored-by: Chavithra PARANA <chavithra@gmail.com> Co-authored-by: Theodore Aptekarev <aptekarev@gmail.com>
102 lines
2.5 KiB
Python
102 lines
2.5 KiB
Python
from typing import Optional
|
|
import pytest
|
|
|
|
try:
|
|
from website import (
|
|
generate_sdk_markdown as gen_sdk,
|
|
generate_terminal_markdown as gen_term,
|
|
)
|
|
except ImportError:
|
|
pytest.skip(allow_module_level=True)
|
|
|
|
|
|
def test_generate_terminal_markdown():
|
|
"""Test the terminal markdown generator"""
|
|
assert gen_term.main() is True
|
|
|
|
|
|
def test_generate_sdk_markdown():
|
|
"""Test the sdk markdown generator"""
|
|
assert gen_sdk.main() is True
|
|
|
|
|
|
def mock_func(arg1: Optional[str] = "Test", arg2: Optional[bool] = True) -> bool:
|
|
"""Stuff here or stuff there, it doesn't matter, it's everywhere.
|
|
|
|
|
|
Parameters
|
|
----------
|
|
arg1 : Optional[str]
|
|
The first argument, by default "Test"
|
|
arg2 : Optional[bool]
|
|
The second argument, by default True
|
|
|
|
Returns
|
|
-------
|
|
bool
|
|
The return value.
|
|
"""
|
|
del arg1, arg2
|
|
return True
|
|
|
|
|
|
# pylint:disable=too-few-public-methods
|
|
class MockTrailMap:
|
|
"""Mock trail map"""
|
|
|
|
def __init__(self):
|
|
self.trailmap = ""
|
|
self.func_attr = {"model": mock_func}
|
|
self.class_attr = {"model": "mock"}
|
|
self.lineon = {"model": 69}
|
|
self.full_path = {"model": "test/mock_func.py"}
|
|
self.model = "mock_func"
|
|
self.long_doc = {"model": mock_func.__doc__}
|
|
self.func_def = {
|
|
"model": 'openbb.mock(arg1: Optional[str] = "Test", arg2: Optional[bool] = True)'
|
|
}
|
|
self.params = {"model": {}}
|
|
for k, p in gen_sdk.get_signature_parameters(
|
|
mock_func, mock_func.__globals__
|
|
).items():
|
|
self.params["model"][k] = p
|
|
|
|
|
|
EXPECTED_OUTPUT = """Stuff here or stuff there, it doesn't matter, it's everywhere.
|
|
|
|
Source Code: [[link](https://github.com/OpenBB-finance/OpenBBTerminal/tree/main/test/mock_func.py#L69)]
|
|
|
|
```python
|
|
openbb.mock(arg1: Optional[str] = "Test", arg2: Optional[bool] = True)
|
|
```
|
|
|
|
---
|
|
|
|
## Parameters
|
|
|
|
| Name | Type | Description | Default | Optional |
|
|
| ---- | ---- | ----------- | ------- | -------- |
|
|
| arg1 | Optional[str] | The first argument, by default "Test" | Test | True |
|
|
| arg2 | Optional[bool] | The second argument, by default True | True | True |
|
|
|
|
|
|
---
|
|
|
|
## Returns
|
|
|
|
| Type | Description |
|
|
| ---- | ----------- |
|
|
| bool | The return value. |
|
|
---
|
|
|
|
"""
|
|
|
|
|
|
def test_sdk_docstring_to_markdown():
|
|
"""Test the docstring to markdown converter"""
|
|
mock = MockTrailMap()
|
|
func_meta = gen_sdk.get_function_meta(mock, "model")
|
|
result = gen_sdk.generate_markdown_section(func_meta)
|
|
|
|
assert result == EXPECTED_OUTPUT
|