Spaces:
Running
Running
| from fasthtml.common import * | |
| from sqlite_minutils.db import Database | |
| import json | |
| import eval_code | |
| def test_some_sqldb_inmem(): | |
| db = Database(memory=True) | |
| assert db is not None | |
| db_tables = db.t | |
| session_states = db_tables.session_states | |
| if session_states not in db.t: | |
| session_states.create(prompt=str, session_id=str, id=int, folder=str, pk='id') | |
| Session_state = session_states.dataclass() | |
| sess = Session_state(prompt="Some prompt", folder="Somewhere") | |
| session_states.insert(sess) | |
| session_states.insert(sess) | |
| session_states.insert(sess) | |
| sess2 = Session_state(prompt="Some prompt", folder="Somewhere new") | |
| # insert AND get the newly inserted ID | |
| g = session_states.insert(sess2) | |
| assert g.id > 3 | |
| g.prompt = "New prompt" | |
| # Update and check if it can be retrieved | |
| session_states.update(g) | |
| # retrieval | |
| newg = session_states(where=f"id={g.id}") | |
| assert len(newg) == 1 | |
| assert newg[0].prompt == "New prompt" | |
| def test_sqlite_jsonops(): | |
| db = Database(memory=True) | |
| ans_js = eval_code.eval_the_piece_of_c_code(None, "") | |
| enh_ans_js = eval_code.add_evaluation_fields_on_js_answer(ans_js) | |
| qa_table = db.t.question_answers.create(id=int, session_id=str, question=str, | |
| answerjs=str, pk="id") | |
| qa_cls = qa_table.dataclass() | |
| qa1 = qa_cls(question="Question", answerjs=enh_ans_js) | |
| assert qa1.id is None | |
| qa1 = qa_table.insert(qa1) | |
| assert qa1.id >= 0 | |
| newans = json.loads(qa1.answerjs) | |
| newans[2]["EVAL"] = eval_code.EVAL_ANSWER_POSITIVE | |
| qa1.answerjs = newans | |
| qa_table.update(qa1) | |
| got_qa = qa_table(where=f"id == {qa1.id}") | |
| assert "EVAL\": 1" in got_qa[0].answerjs | |