Skip to content

Commit 912b471

Browse files
authored
Update DataFeederSpec.scala
1 parent 44a2ef9 commit 912b471

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/test/scala/common/DataFeederSpec.scala

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ import org.scalatest.matchers.should.Matchers
99
class DataFeederSpec extends AnyFlatSpec with Matchers {
1010

1111
"NumpyDataFeeder" should "feed X-ray data correctly" in {
12-
val filename = "25-trimmed.npy"
12+
val filename = "test_data/25-trimmed.npy"
1313

1414
// Test feeding data as stream
1515
val dataStream = NumpyDataFeeder.feedNumpyData(filename)
@@ -33,7 +33,7 @@ class DataFeederSpec extends AnyFlatSpec with Matchers {
3333
}
3434

3535
"NumpyDataFeeder" should "feed data in chunks" in {
36-
val filename = "25-trimmed.npy"
36+
val filename = "test_data/25-trimmed.npy"
3737
val chunkSize = 1024
3838

3939
val chunkIterator = NumpyDataFeeder.feedNumpyDataInChunks(filename, chunkSize)
@@ -78,7 +78,7 @@ class DataFeederSpec extends AnyFlatSpec with Matchers {
7878
}
7979

8080
"NumpyDataFeeder" should "handle zero patterns correctly" in {
81-
val filename = "25-trimmed.npy"
81+
val filename = "test_data/25-trimmed.npy"
8282

8383
val dataStream = NumpyDataFeeder.feedNumpyData(filename)
8484
val firstHundred = dataStream.take(100).toArray
@@ -92,4 +92,4 @@ class DataFeederSpec extends AnyFlatSpec with Matchers {
9292
// Should have significant zeros (X-ray data characteristic)
9393
zeroRatio should be > 0.5
9494
}
95-
}
95+
}

0 commit comments

Comments
 (0)