Skip to content

Commit a0469d6

Browse files
author
kx79wq
committed
test: try fixing spark histogram names
1 parent 0177902 commit a0469d6

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

histogrammar/dfinterface/spark_histogrammar.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -253,6 +253,7 @@ def fill_histogram(self, idf, features):
253253

254254
# do the actual filling
255255
hist.fill.sparksql(idf)
256+
hist.name = name
256257
self._hists[name] = hist
257258

258259
def _execute(self, df):

tests/test_spark_histogrammar.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def test_get_histograms(spark_co):
113113
@pytest.mark.skipif(not spark_found, reason="spark not found")
114114
@pytest.mark.filterwarnings("ignore:createDataFrame attempted Arrow optimization because")
115115
def test_get_histograms_module(spark_co):
116-
pytest.age["data"]["name"] = "'age'"
116+
pytest.age["data"]["name"] = "age"
117117
pytest.company["data"]["name"] = "'company'"
118118
pytest.eyesColor["data"]["name"] = "'eyeColor'"
119119
pytest.gender["data"]["name"] = "'gender'"

0 commit comments

Comments
 (0)