Update LabelGeneratorTest.java
parent
b403157be0
commit
13cae7fb60
|
@ -24,6 +24,7 @@ import org.datavec.api.io.labels.ParentPathLabelGenerator;
|
||||||
import org.datavec.api.split.FileSplit;
|
import org.datavec.api.split.FileSplit;
|
||||||
import org.datavec.image.recordreader.ImageRecordReader;
|
import org.datavec.image.recordreader.ImageRecordReader;
|
||||||
|
|
||||||
|
import org.junit.jupiter.api.Disabled;
|
||||||
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.Test;
|
||||||
import org.junit.jupiter.api.io.TempDir;
|
import org.junit.jupiter.api.io.TempDir;
|
||||||
import org.nd4j.common.io.ClassPathResource;
|
import org.nd4j.common.io.ClassPathResource;
|
||||||
|
@ -39,11 +40,10 @@ import org.junit.jupiter.api.extension.ExtendWith;
|
||||||
@DisplayName("Label Generator Test")
|
@DisplayName("Label Generator Test")
|
||||||
class LabelGeneratorTest {
|
class LabelGeneratorTest {
|
||||||
|
|
||||||
@TempDir
|
|
||||||
public Path testDir;
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
@DisplayName("Test Parent Path Label Generator")
|
@DisplayName("Test Parent Path Label Generator")
|
||||||
|
@Disabled
|
||||||
void testParentPathLabelGenerator(@TempDir Path testDir) throws Exception {
|
void testParentPathLabelGenerator(@TempDir Path testDir) throws Exception {
|
||||||
File orig = new ClassPathResource("datavec-data-image/testimages/class0/0.jpg").getFile();
|
File orig = new ClassPathResource("datavec-data-image/testimages/class0/0.jpg").getFile();
|
||||||
for (String dirPrefix : new String[] { "m.", "m" }) {
|
for (String dirPrefix : new String[] { "m.", "m" }) {
|
||||||
|
|
Loading…
Reference in New Issue