Skip to content

Commit 47a8564

Browse files
committed
2 parents e5e436c + f4cbb52 commit 47a8564

4 files changed

Lines changed: 8 additions & 8 deletions

File tree

test/test_convert_to_numeric.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import pandas as pd
2-
from to_numeric import convert_to_numeric
2+
from toolkit.data_processing import convert_to_numeric
33

44
def test_convert_to_numeric():
55

@@ -9,4 +9,4 @@ def test_convert_to_numeric():
99
convert_to_numeric(df, 'A')
1010

1111

12-
assert df['A'][0],int
12+
assert df['A'][0],int

test/test_heatmap_corr.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,12 @@
22
import matplotlib.pyplot as plt
33
import seaborn as sns
44
import pandas as pd
5-
from heatmap_corr import heatmap
5+
from toolkit.plot import heatmap
66

77
def test_heatmap():
88
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9], 'target': [10, 20, 30]})
99

1010
heatmap(df, 2, 'target', None)
1111

1212

13-
13+

test/test_scrap.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import time
88
import io
99
from PIL import Image
10-
from scrap import image_scrap
10+
from toolkit.machine_learning import image_scrap
1111
import shutil
1212

1313

@@ -21,4 +21,4 @@ def test_image_scrap():
2121
assert os.path.exists(download_dir)
2222
assert len(os.listdir(download_dir)) == n
2323

24-
shutil.rmtree(download_dir)
24+
shutil.rmtree(download_dir)

test/test_sunburst.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from sunburst import sunburst
1+
from toolkit.plot import sunburst
22
import plotly.graph_objs as go
33
import plotly.express as px
44

@@ -12,4 +12,4 @@ def test_sunburst():
1212

1313
assert fig.layout.title.text == 'My Sunburst Chart'
1414
assert fig.layout.width == 800
15-
assert fig.layout.height == 600
15+
assert fig.layout.height == 600

0 commit comments

Comments
 (0)