Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
S
spark
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Model registry
Operate
Environments
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
cs525-sp18-g07
spark
Commits
0f4fbb05
Commit
0f4fbb05
authored
12 years ago
by
Shivaram Venkataraman
Browse files
Options
Downloads
Patches
Plain Diff
Change BlockManagerSuite test cases to use a deterministic size estimator and
update the results to match the new estimates
parent
22ba3a3f
No related branches found
Branches containing commit
No related tags found
Tags containing commit
No related merge requests found
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
core/src/test/scala/spark/storage/BlockManagerSuite.scala
+31
-9
31 additions, 9 deletions
core/src/test/scala/spark/storage/BlockManagerSuite.scala
with
31 additions
and
9 deletions
core/src/test/scala/spark/storage/BlockManagerSuite.scala
+
31
−
9
View file @
0f4fbb05
...
@@ -6,17 +6,27 @@ import akka.actor._
...
@@ -6,17 +6,27 @@ import akka.actor._
import
org.scalatest.FunSuite
import
org.scalatest.FunSuite
import
org.scalatest.BeforeAndAfter
import
org.scalatest.BeforeAndAfter
import
org.scalatest.PrivateMethodTester
import
spark.KryoSerializer
import
spark.KryoSerializer
import
spark.SizeEstimator
import
spark.util.ByteBufferInputStream
import
spark.util.ByteBufferInputStream
class
BlockManagerSuite
extends
FunSuite
with
BeforeAndAfter
{
class
BlockManagerSuite
extends
FunSuite
with
BeforeAndAfter
with
PrivateMethodTester
{
var
actorSystem
:
ActorSystem
=
null
var
actorSystem
:
ActorSystem
=
null
var
master
:
BlockManagerMaster
=
null
var
master
:
BlockManagerMaster
=
null
var
oldArch
:
String
=
_
var
oldOops
:
String
=
_
before
{
before
{
actorSystem
=
ActorSystem
(
"test"
)
actorSystem
=
ActorSystem
(
"test"
)
master
=
new
BlockManagerMaster
(
actorSystem
,
true
,
true
)
master
=
new
BlockManagerMaster
(
actorSystem
,
true
,
true
)
// Set the arch to 64-bit and compressedOops to true to get a deterministic test-case
oldArch
=
System
.
setProperty
(
"os.arch"
,
"amd64"
)
oldOops
=
System
.
setProperty
(
"spark.test.useCompressedOops"
,
"true"
)
val
initialize
=
PrivateMethod
[
Unit
](
'initialize
)
SizeEstimator
invokePrivate
initialize
()
}
}
after
{
after
{
...
@@ -24,6 +34,18 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -24,6 +34,18 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
actorSystem
.
awaitTermination
()
actorSystem
.
awaitTermination
()
actorSystem
=
null
actorSystem
=
null
master
=
null
master
=
null
if
(
oldArch
!=
null
)
{
System
.
setProperty
(
"os.arch"
,
oldArch
)
}
else
{
System
.
clearProperty
(
"os.arch"
)
}
if
(
oldOops
!=
null
)
{
System
.
setProperty
(
"spark.test.useCompressedOops"
,
oldOops
)
}
else
{
System
.
clearProperty
(
"spark.test.useCompressedOops"
)
}
}
}
test
(
"manager-master interaction"
)
{
test
(
"manager-master interaction"
)
{
...
@@ -57,7 +79,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -57,7 +79,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
}
}
test
(
"in-memory LRU storage"
)
{
test
(
"in-memory LRU storage"
)
{
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
0
00
)
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
2
00
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
...
@@ -78,7 +100,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -78,7 +100,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
}
}
test
(
"in-memory LRU storage with serialization"
)
{
test
(
"in-memory LRU storage with serialization"
)
{
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
0
00
)
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
2
00
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
...
@@ -99,7 +121,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -99,7 +121,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
}
}
test
(
"on-disk storage"
)
{
test
(
"on-disk storage"
)
{
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
0
00
)
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
2
00
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
...
@@ -112,7 +134,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -112,7 +134,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
}
}
test
(
"disk and memory storage"
)
{
test
(
"disk and memory storage"
)
{
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
0
00
)
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
2
00
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
...
@@ -126,7 +148,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -126,7 +148,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
}
}
test
(
"disk and memory storage with serialization"
)
{
test
(
"disk and memory storage with serialization"
)
{
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
0
00
)
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
2
00
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
...
@@ -140,7 +162,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -140,7 +162,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
}
}
test
(
"LRU with mixed storage levels"
)
{
test
(
"LRU with mixed storage levels"
)
{
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
0
00
)
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
2
00
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a1
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a2
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
val
a3
=
new
Array
[
Byte
](
400
)
...
@@ -166,7 +188,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -166,7 +188,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
}
}
test
(
"in-memory LRU with streams"
)
{
test
(
"in-memory LRU with streams"
)
{
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
0
00
)
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
2
00
)
val
list1
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list1
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list2
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list2
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list3
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list3
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
...
@@ -192,7 +214,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
...
@@ -192,7 +214,7 @@ class BlockManagerSuite extends FunSuite with BeforeAndAfter {
}
}
test
(
"LRU with mixed storage levels and streams"
)
{
test
(
"LRU with mixed storage levels and streams"
)
{
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
0
00
)
val
store
=
new
BlockManager
(
master
,
new
KryoSerializer
,
1
2
00
)
val
list1
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list1
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list2
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list2
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list3
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
val
list3
=
List
(
new
Array
[
Byte
](
200
),
new
Array
[
Byte
](
200
))
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment