MarcoEngine/tests.py at main · MarcoNITE/MarcoEngine · GitHub

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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import python_checking
import pytest
import requests
import zipfile
import os
import sys
import shutil
import stat
import chess
import chess.engine
import train

platform = sys.platform
file_extension = ".exe" if platform == "win32" else ""

def test_nothing():
    assert True

def check_os():
    if platform == "linux":
        assert True
        return

    else:
        assert False

def download_me():
    response = requests.get("https://github.com/MarcoNITE/MarcoEngine/archive/master.zip", allow_redirects=True)
    with open("master.zip", "wb") as file:
        file.write(response.content)
    with zipfile.ZipFile("master.zip", "r") as zip_ref:
        zip_ref.extractall(".")

def delete_downloaded_zip():
    os.remove("master.zip")

def download_sf():
    windows_or_linux = "win" if platform == "win32" else "linux"
    response = requests.get(f"https://stockfishchess.org/files/stockfish_14.1_{windows_or_linux}_x64.zip", allow_redirects=True)
    with open("./TEMP/sf_zip.zip", "wb") as file:
        file.write(response.content)
    with zipfile.ZipFile("./TEMP/sf_zip.zip", "r") as zip_ref:
        zip_ref.extractall("./TEMP/")
    shutil.copyfile(f"./TEMP/stockfish_14.1_{windows_or_linux}_x64/stockfish_14.1_{windows_or_linux}_x64{file_extension}", f"./TEMP/sf{file_extension}")
    shutil.copyfile(f"./TEMP/sf{file_extension}", f"./TEMP/sf2{file_extension}")
    if windows_or_linux == "linux":
        st = os.stat(f"./TEMP/sf{file_extension}")
        os.chmod(f"./TEMP/sf{file_extension}", st.st_mode | stat.S_IEXEC)
        st = os.stat(f"./TEMP/sf2{file_extension}")
        os.chmod(f"./TEMP/sf2{file_extension}", st.st_mode | stat.S_IEXEC)


if os.path.exists("TEMP"):
    shutil.rmtree("TEMP")
os.mkdir("TEMP")

check_os()
download_me()
delete_downloaded_zip()
download_sf()