Skip to content

Commit 038662c

Browse files
committed
don't bother mocking
1 parent b266635 commit 038662c

1 file changed

Lines changed: 5 additions & 8 deletions

File tree

tests/integration/test_ctgan.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,6 @@
99
model are not checked.
1010
"""
1111

12-
from unittest.mock import patch
13-
1412
import numpy as np
1513
import pandas as pd
1614

@@ -53,12 +51,11 @@ def test_ctgan_numpy():
5351

5452

5553
def test_log_frequency():
56-
with patch('numpy.random.choice',
57-
return_value=np.repeat(['a', 'b', 'c'], [950, 25, 25])):
58-
data = pd.DataFrame({
59-
'continuous': np.random.random(1000),
60-
'discrete': np.random.choice(['a', 'b', 'c'], 1000, p=[0.95, 0.025, 0.025])
61-
})
54+
55+
data = pd.DataFrame({
56+
'continuous': np.random.random(1000),
57+
'discrete': np.repeat(['a', 'b', 'c'], [950, 25, 25])
58+
})
6259

6360
discrete_columns = ['discrete']
6461

0 commit comments

Comments
 (0)