Skip to content

Commit

Permalink
adapt shared memory test to github actions (only 1 core available)
Browse files Browse the repository at this point in the history
  • Loading branch information
@SimonBlanke
SimonBlanke committed Aug 29, 2021
1 parent e1d5f53 commit c531a12
Showing 1 changed file with 35 additions and 18 deletions.
53 changes: 35 additions & 18 deletions tests/test_memory/test_shared_memory.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -68,36 +68,40 @@ def model4(opt):

def test_shared_memory_0():
c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
n_iter=100,
n_jobs=1,
memory=True,
)
hyper.run()
d_time_1 = time.time() - c_time

c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
n_iter=100,
n_jobs=4,
memory=True,
)
hyper.run()
d_time_2 = time.time() - c_time

print("\n d_time_1 \n", d_time_1)
print("\n d_time_2 \n", d_time_2)

assert d_time_2 * 1.4 - d_time_1 < 0
d_time_2 = d_time_2 / 2

assert d_time_2 - d_time_1 < 0


def test_shared_memory_1():
c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand All @@ -108,7 +112,7 @@ def test_shared_memory_1():
d_time_1 = time.time() - c_time

c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand All @@ -121,12 +125,14 @@ def test_shared_memory_1():
print("\n d_time_1 \n", d_time_1)
print("\n d_time_2 \n", d_time_2)

d_time_2 = d_time_2 / 2

assert d_time_1 / d_time_2 > 1.1


def test_shared_memory_2():
c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand All @@ -137,7 +143,7 @@ def test_shared_memory_2():
d_time_1 = time.time() - c_time

c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand Down Expand Up @@ -168,12 +174,14 @@ def test_shared_memory_2():
print("\n d_time_1 \n", d_time_1)
print("\n d_time_2 \n", d_time_2)

d_time_2 = d_time_2 / 2

assert d_time_1 / d_time_2 > 1.1


def test_shared_memory_3():
c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model1,
search_space,
Expand All @@ -190,7 +198,7 @@ def test_shared_memory_3():
d_time_1 = time.time() - c_time

c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand Down Expand Up @@ -227,13 +235,14 @@ def test_shared_memory_3():
print("\n d_time_1 \n", d_time_1)
print("\n d_time_2 \n", d_time_2)

d_time_2 = d_time_2 / 2

assert d_time_1 / d_time_2 < 1.2
assert d_time_1 / d_time_2 > 0.8


def test_shared_memory_warm_start_0():
c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand All @@ -246,7 +255,7 @@ def test_shared_memory_warm_start_0():
search_data0 = hyper.results(model)

c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand All @@ -260,12 +269,14 @@ def test_shared_memory_warm_start_0():
print("\n d_time_1 \n", d_time_1)
print("\n d_time_2 \n", d_time_2)

d_time_2 = d_time_2 / 2

assert d_time_2 * 1.4 - d_time_1 < 0


def test_shared_memory_warm_start_1():
c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand All @@ -278,7 +289,7 @@ def test_shared_memory_warm_start_1():
search_data0 = hyper.results(model)

c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand All @@ -292,12 +303,14 @@ def test_shared_memory_warm_start_1():
print("\n d_time_1 \n", d_time_1)
print("\n d_time_2 \n", d_time_2)

d_time_2 = d_time_2 / 2

assert d_time_1 / d_time_2 > 1.1


def test_shared_memory_warm_start_2():
c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand All @@ -310,7 +323,7 @@ def test_shared_memory_warm_start_2():
search_data0 = hyper.results(model)

c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand Down Expand Up @@ -345,12 +358,14 @@ def test_shared_memory_warm_start_2():
print("\n d_time_1 \n", d_time_1)
print("\n d_time_2 \n", d_time_2)

d_time_2 = d_time_2 / 2

assert d_time_1 / d_time_2 > 1.1


def test_shared_memory_warm_start_3():
c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand Down Expand Up @@ -387,7 +402,7 @@ def test_shared_memory_warm_start_3():
search_data0 = hyper.results(model1)

c_time = time.time()
hyper = Hyperactive()
hyper = Hyperactive(n_processes=1)
hyper.add_search(
model,
search_space,
Expand Down Expand Up @@ -429,4 +444,6 @@ def test_shared_memory_warm_start_3():
print("\n d_time_1 \n", d_time_1)
print("\n d_time_2 \n", d_time_2)

d_time_2 = d_time_2 / 2

assert d_time_1 / d_time_2 > 1.4

0 comments on commit c531a12

Please sign in to comment.