Tests: Update tests

lucianolorenti · Jun 16, 2024 · 95c6ff3 · 95c6ff3
1 parent 12d1192
commit 95c6ff3
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/docs/dataset/analysis/Sensor Validation.ipynb b/docs/dataset/analysis/Sensor Validation.ipynb
@@ -2791,7 +2791,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.9"
+   "version": "3.11.7"
   }
  },
  "nbformat": 4,

diff --git a/tests/test_analysis.py b/tests/test_analysis.py
@@ -1,10 +1,9 @@
 
 
-from random import sample
-from typing import Dict
+
 import numpy as np
 import pandas as pd
-from ceruleo.dataset.analysis.numerical_features import NumericalFeaturesAnalysis, analysis
+from ceruleo.dataset.analysis.numerical_features import NumericalFeaturesAnalysis, analyze
 
 from ceruleo.dataset.ts_dataset import AbstractPDMDataset
 from ceruleo.dataset.analysis.correlation import CorrelationAnalysis, correlation_analysis
@@ -104,6 +103,6 @@ def test_distribution(self):
 
     def test_analysis(self):
         dataset = MockDataset(5)
-        df = analysis(dataset)
+        df = analyze(dataset)
         keys = list(df.keys())
         assert isinstance(df[keys[0]], NumericalFeaturesAnalysis)
diff --git a/tests/test_graphics.py b/tests/test_graphics.py
@@ -181,5 +181,5 @@ def test_durations_boxplot(self):
     )
     def test_durations_histogram(self):
         dataset = MockDataset(7)
-        ax = durations_histogram(dataset)
+        ax = durations_histogram(dataset, label="Test Dataset")
         return ax.figure