data tests run the code

cpmpercussion · Jun 26, 2024 · 6565f1d · 6565f1d
1 parent 268d334
commit 6565f1d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/impsy/tests/test_data.py b/impsy/tests/test_data.py
@@ -1,4 +1,5 @@
 from impsy import dataset
+import numpy as np
 import os
 
 
@@ -8,11 +9,12 @@ def test_log_to_examples():
   test_line = "2024-06-01T12:00:00,interface,0.1,0.2,0.3,0.4"
   with open(test_log_file, "w") as file:
     file.write(test_line)
-  dataset.transform_log_to_sequence_example(test_log_file, 4)
+  log = dataset.transform_log_to_sequence_example(test_log_file, 4)
+  assert(isinstance(log, np.ndarray))
   os.remove(test_log_file)
 
 
 def test_dataset_command():
-  """Test the dataset command"""
+  """Test the dataset command runs"""
   print("Testing dataset function...")
   dataset.generate_dataset(dimension=2, source="logs")