Skip to content

Commit 8018270

Browse files
correcting the name of the column for issue #182 in both the scala and the python code
1 parent cc6d2ae commit 8018270

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

python/tests/tsdf_tests.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def test_describe(self):
156156
# joined dataframe should equal the expected dataframe
157157
# self.assertDataFramesEqual(res, dfExpected)
158158
assert res.count() == 7
159-
assert res.filter(F.col("unique_ts_count") != " ").select(F.max(F.col('unique_ts_count'))).collect()[0][
159+
assert res.filter(F.col("unique_time_series_count") != " ").select(F.max(F.col('unique_time_series_count'))).collect()[0][
160160
0] == "1"
161161
assert res.filter(F.col("min_ts") != " ").select(F.col('min_ts').cast("string")).collect()[0][
162162
0] == '2020-08-01 00:00:10'

scala/tempo/src/main/scala/com/databrickslabs/tempo/TSDF.scala

+1
Original file line numberDiff line numberDiff line change
@@ -453,6 +453,7 @@ private[tempo] sealed class BaseTSDF(val df: DataFrame,
453453
val global_smry_rec = desc_stats.limit(1).select(non_summary_cols_blank: _*)
454454

455455
val full_smry = global_smry_rec.union(desc_stats)
456+
val full_smry = full_smry.withColumnRenamed("unique_ts_count","unique_time_series_count")
456457

457458
return(full_smry)
458459
}

scala/tempo/src/test/scala/com/databrickslabs/tempo/DescribeTests.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ class DescribeTests
4242
// joined dataframe should equal the expected dataframe
4343
// self.assertDataFramesEqual(res, dfExpected)
4444
assert(res.count == 7)
45-
assert(res.select(max(col("unique_ts_count"))).collect()(0)(0) == "1")
45+
assert(res.select(max(col("unique_time_series_count"))).collect()(0)(0) == "1")
4646
assert(res.select(col("min_ts").cast("string")).collect()(0)(0) == "2020-08-01 00:00:10")
4747
assert(res.select(col("max_ts").cast("string")).collect()(0)(0) == "2020-09-01 00:19:12")
4848
}

0 commit comments

Comments
 (0)