Skip to content

Commit 19c31af

Browse files
author
Ilya Ganelin
committed
Added cleaner computation of time conversions in tests
1 parent 6387772 commit 19c31af

File tree

1 file changed

+14
-12
lines changed

1 file changed

+14
-12
lines changed

core/src/test/scala/org/apache/spark/util/UtilsSuite.scala

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717

1818
package org.apache.spark.util
1919

20+
import java.util.concurrent.TimeUnit
21+
2022
import scala.util.Random
2123

2224
import java.io.{File, ByteArrayOutputStream, ByteArrayInputStream, FileOutputStream}
@@ -40,25 +42,25 @@ class UtilsSuite extends FunSuite with ResetSystemProperties {
4042
assert(Utils.timeStringAsS("1s") === 1)
4143
assert(Utils.timeStringAsS("1000ms") === 1)
4244
assert(Utils.timeStringAsS("1000000us") === 1)
43-
assert(Utils.timeStringAsS("1min") === 60)
44-
assert(Utils.timeStringAsS("1h") === 60 * 60)
45-
assert(Utils.timeStringAsS("1d") === 60 * 60 * 24)
45+
assert(Utils.timeStringAsS("1min") === TimeUnit.MINUTES.toSeconds(1))
46+
assert(Utils.timeStringAsS("1h") === TimeUnit.HOURS.toSeconds(1))
47+
assert(Utils.timeStringAsS("1d") === TimeUnit.DAYS.toSeconds(1))
4648

4749
assert(Utils.timeStringAsMs("1") === 1)
4850
assert(Utils.timeStringAsMs("1ms") === 1)
4951
assert(Utils.timeStringAsMs("1000us") === 1)
50-
assert(Utils.timeStringAsMs("1s") === 1000)
51-
assert(Utils.timeStringAsMs("1min") === 1000 * 60)
52-
assert(Utils.timeStringAsMs("1h") === 1000 * 60 * 60)
53-
assert(Utils.timeStringAsMs("1d") === 1000 * 60 * 60 * 24)
52+
assert(Utils.timeStringAsMs("1s") === TimeUnit.SECONDS.toMillis(1))
53+
assert(Utils.timeStringAsMs("1min") === TimeUnit.MINUTES.toMillis(1))
54+
assert(Utils.timeStringAsMs("1h") === TimeUnit.HOURS.toMillis(1))
55+
assert(Utils.timeStringAsMs("1d") === TimeUnit.DAYS.toMillis(1))
5456

5557
assert(Utils.timeStringAsUs("1") === 1)
5658
assert(Utils.timeStringAsUs("1us") === 1)
57-
assert(Utils.timeStringAsUs("1ms") === 1000)
58-
assert(Utils.timeStringAsUs("1s") === 1000 * 1000)
59-
assert(Utils.timeStringAsUs("1min") === 1000 * 1000 * 60)
60-
assert(Utils.timeStringAsUs("1h") === 1000L * 1000L * 60 * 60)
61-
assert(Utils.timeStringAsUs("1d") === 1000L * 1000L * 60 * 60 * 24)
59+
assert(Utils.timeStringAsUs("1ms") === TimeUnit.MILLISECONDS.toMillis(1))
60+
assert(Utils.timeStringAsUs("1s") === TimeUnit.SECONDS.toSeconds(1))
61+
assert(Utils.timeStringAsUs("1min") === TimeUnit.MINUTES.toMicros(1))
62+
assert(Utils.timeStringAsUs("1h") === TimeUnit.HOURS.toMicros(1))
63+
assert(Utils.timeStringAsUs("1d") === TimeUnit.DAYS.toMicros(1))
6264
}
6365

6466
test("bytesToString") {

0 commit comments

Comments
 (0)