1+ from requests import get
2+ from unittest import TestCase , main
3+ try :
4+ from .find_primes import *
5+
6+ except ImportError :
7+ from find_primes import *
8+
9+ def get_answer (string ):
10+ try :
11+ t = get (f'https://oeis.org/{ string } ' ).text
12+
13+ except Exception :
14+ try :
15+ t = get (f'https://oeis.org/{ string } ' ).text
16+
17+ except Exception :
18+ return
19+
20+ located_1 = t .find ('<tt>' )
21+ located_2 = t .find ('</tt>' )
22+ answer_str = t [located_1 :located_2 ][4 :]
23+ return [int (x ) for x in answer_str .split (', ' )]
24+
25+ class TestAnswer (TestCase ):
26+ def test_all_prime (self ):
27+ result = all_primes (275 , 'list' )
28+ answer = get_answer ('A000040' )
29+ self .assertEqual (result , answer )
30+
31+ def test_twin_prime (self ):
32+ result = find_twins (1610 )
33+ answer1 = get_answer ('A001359' )
34+ answer2 = get_answer ('A006512' )
35+ self .assertEqual (list (result .keys ()), answer1 )
36+ self .assertEqual (list (result .values ()), answer2 )
37+
38+ def test_palindome_prime (self ):
39+ result = find_palindromes (18190 )
40+ answer = get_answer ('A002385' )
41+ self .assertEqual (result , answer )
42+
43+ if __name__ == '__main__' :
44+ main ()
0 commit comments