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
|
import os
import pytest
from mp_api.client.routes.materials.robocrys import RobocrysRester
@pytest.fixture
def rester():
rester = RobocrysRester()
yield rester
rester.session.close()
@pytest.mark.skipif(os.getenv("MP_API_KEY", None) is None, reason="No API key found.")
def test_client(rester):
search_method = rester.search
if search_method is not None:
q = {"keywords": ["silicon"], "num_chunks": 1}
doc = search_method(**q)[0]
assert doc.description is not None
assert doc.condensed_structure is not None
|
