|
|
@@ -28,7 +28,10 @@ def setup_module(): |
|
|
|
recursive_unlink(testdb) |
|
|
|
|
|
|
|
# Start web app on a custom port |
|
|
|
test_db = nilmdb.utils.serializer_proxy(nilmdb.server.NilmDB)(testdb) |
|
|
|
test_db = nilmdb.utils.serializer_proxy(nilmdb.server.NilmDB)( |
|
|
|
testdb, bulkdata_args = { "file_size" : 16384, |
|
|
|
"files_per_dir" : 3 } ) |
|
|
|
|
|
|
|
test_server = nilmdb.server.Server(test_db, host = "127.0.0.1", |
|
|
|
port = 32180, stoppable = False, |
|
|
|
fast_shutdown = True, |
|
|
@@ -179,6 +182,17 @@ class TestNumpyClient(object): |
|
|
|
assert(np.array_equal(a,b)) |
|
|
|
assert(np.array_equal(a,c)) |
|
|
|
|
|
|
|
# Make sure none of the files are greater than 16384 bytes as |
|
|
|
# we configured with the bulkdata_args above. |
|
|
|
datapath = os.path.join(testdb, "data") |
|
|
|
for (dirpath, dirnames, filenames) in os.walk(datapath): |
|
|
|
for f in filenames: |
|
|
|
fn = os.path.join(dirpath, f) |
|
|
|
size = os.path.getsize(fn) |
|
|
|
if size > 16384: |
|
|
|
raise AssertionError(sprintf("%s is too big: %d > %d\n", |
|
|
|
fn, size, 16384)) |
|
|
|
|
|
|
|
nilmdb.client.numpyclient.StreamInserterNumpy._max_data = old_max_data |
|
|
|
client.close() |
|
|
|
|
|
|
|