This repository was archived by the owner on Aug 20, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathnoxfile.py
61 lines (51 loc) · 1.8 KB
/
noxfile.py
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
61
import nox
@nox.session()
@nox.parametrize(
"python,pyspark",
[
(python, pyspark)
for python in ("3.7", "3.8")
for pyspark in ("2.4.5", "3.0.1","3.1.1")
if (python, pyspark) != ("3.8", "2.4.5")
],)
def tests(session, pyspark):
session.run("python", "-m", "pip", "install", "--upgrade", "pip")
session.install("cmake")
session.install("pytest")
session.install("pytest-order")
session.install("pandas==0.25.3")
session.install("numpy==1.19.5", "--no-deps")
session.install("scipy==1.6.0", "--no-deps")
session.install("node2vec==0.4.3")
session.install("graphframes==0.6.0")
if pyspark == "2.4.5":
session.install("pyarrow==0.14.1", "--no-deps")
else:
session.install("pyarrow==2.0.0", "--no-deps")
session.install("networkx==2.5.1")
session.install(f"pyspark=={pyspark}")
session.run("pytest", "-x","-v", "-W", "ignore::DeprecationWarning")
@nox.session(python="3.6")
@nox.parametrize(
"pyspark",
[
pyspark
for pyspark in ("2.4.5", "3.0.1","3.1.1")
],)
def tests_in_3_6(session, pyspark):
session.run("python", "-m", "pip", "install", "--upgrade", "pip")
session.install("cmake")
session.install("pytest")
session.install("pytest-order")
session.install("pandas==0.25.3")
session.install("numpy==1.19.5", "--no-deps")
session.install("scipy==1.5.4", "--no-deps")
session.install("node2vec==0.4.3")
session.install("graphframes==0.6.0")
if pyspark == "2.4.5":
session.install("pyarrow==0.14.1", "--no-deps")
else:
session.install("pyarrow==2.0.0", "--no-deps")
session.install("networkx==2.5.1")
session.install(f"pyspark=={pyspark}")
session.run("pytest", "-x","-v", "-W", "ignore::DeprecationWarning")