|
@@ 206-220 (lines=15) @@
|
| 203 |
|
assert results1.equals(results2) |
| 204 |
|
|
| 205 |
|
|
| 206 |
|
def test_ltm_float(): |
| 207 |
|
path = "./" |
| 208 |
|
model_name = "test_ltm_float" |
| 209 |
|
array = np.arange(3 * 10).reshape(10, 3) |
| 210 |
|
array = array / 1000 |
| 211 |
|
results1 = pd.DataFrame(array, columns=["x1", "x2", "x3"]) |
| 212 |
|
|
| 213 |
|
memory = LongTermMemory(model_name, path=path) |
| 214 |
|
memory.save(results1, objective_function) |
| 215 |
|
|
| 216 |
|
results2 = memory.load() |
| 217 |
|
|
| 218 |
|
memory.remove_model_data() |
| 219 |
|
|
| 220 |
|
assert results1.equals(results2) |
| 221 |
|
|
| 222 |
|
|
| 223 |
|
def test_ltm_str(): |
|
@@ 190-203 (lines=14) @@
|
| 187 |
|
""" |
| 188 |
|
|
| 189 |
|
|
| 190 |
|
def test_ltm_int(): |
| 191 |
|
path = "./" |
| 192 |
|
model_name = "test_ltm_int" |
| 193 |
|
array = np.arange(3 * 10).reshape(10, 3) |
| 194 |
|
results1 = pd.DataFrame(array, columns=["x1", "x2", "x3"]) |
| 195 |
|
|
| 196 |
|
memory = LongTermMemory(model_name, path=path) |
| 197 |
|
memory.save(results1, objective_function) |
| 198 |
|
|
| 199 |
|
results2 = memory.load() |
| 200 |
|
|
| 201 |
|
memory.remove_model_data() |
| 202 |
|
|
| 203 |
|
assert results1.equals(results2) |
| 204 |
|
|
| 205 |
|
|
| 206 |
|
def test_ltm_float(): |