test_solver.py

import pytest
from solver import TimeLockPuzzleSolver


def test_run():
    assert TimeLockPuzzleSolver(
        N=146316494799812047511623076708218946420205748539793012358172922720474915105527936143913152868234855653659362702149685726946497663225575817296009946943300164053238818934292739741516716926800148512198028885552295620099510315701145312492185132376573874729991127362677325304713690683683766343670797779445517035741,
        A=2,
        T=800000
    ).run() == 86760125054422556897363885542358145618076815220095386261325558414510076989927762880935623265951835010937120114742503498144946310658675917071466485601394098486420576909352992488044505145523618712937783100889950928662334399016911369464456878404586985118054482283245971516263455232529136213962251072935371672855

    assert TimeLockPuzzleSolver(
        N=153526702603894411080057434083359993508379431385313068673214353678785110519166073488278522921096111604692560956116181458932958229499055081025670022217203757970233347943436262619280966431517401313566817574599449508613578189620321056046057045462186230874709780746979596374602946215650155219803602360524947391093,
        A=2,
        T=800000
    ).run() == 17762742375447065679724090887366591373247150941676661073878151931208580646621232357476946880110037860360431468460303140487632939089842755853137914445666261117465775213643793289161121437110649043056427207087284572749538739144018882183490982033011528028999489874150199702105774653066960479825514953355087467578