def test_training_zerobatch(mocked_config):
TestTrainingModule.batch_size = 0
mocked_config.side_effect = TestTrainingModule.parse
with pytest.raises(ZeroDivisionError):
TrainingModule.train(
"modules/test/unittest/model/training_module/test.hdf5", "",
"", "modules/test/unittest/model/training_module/", 0.5)
def test_training_zeroconv(mocked_config):
TestTrainingModule.num_conv_layers = 0
mocked_config.side_effect = TestTrainingModule.parse
with pytest.raises(RuntimeError):
TrainingModule.train(
"modules/test/unittest/model/training_module/test.hdf5", "",
"", "modules/test/unittest/model/training_module/", 0.5)
def execute(self):
super().execute()
TrainingModule.train(
matrices_path=self.arguments.get(Key.PATH),
neural_network_path=self.arguments.get(Key.EXISTING_NETWORK),
name=self.arguments.get(Key.NAME),
saving_path=self.arguments.get(Key.SAVING_PATH),
)
def execute(self):
super().execute()
TrainingModule.train(
matrices_path=self.arguments.get(Key.PATH),
neural_network_path=self.arguments.get(Key.NETWORK),
name=self.arguments.get(Key.NAME),
saving_path=self.arguments.get(Key.SAVING_PATH),
training_test_split=self.get_float_value(Key.TRAIN),
)
def test_training_createfile(mocked_config):
mocked_config.side_effect = TestTrainingModule.parse
TrainingModule.train(
"modules/test/unittest/model/training_module/test.hdf5", "", "",
"modules/test/unittest/model/training_module/", 0.5)
file_found = False
for file in os.listdir("modules/test/unittest/model/training_module/"):
if str(file)[:2] == "01" and str(file)[-4:] == "hdf5":
file_found = True
assert file_found
def __register_output_service(self):
Collector.set_output_service(self.__output_service)
LabelingModule.set_output_service(self.__output_service)
TrainingModule.set_output_service(self.__output_service)
Classifier.set_output_service(self.__output_service)
def test_training_zerodense(mocked_config):
TestTrainingModule.num_dense_layers = 0
mocked_config.side_effect = TestTrainingModule.parse
TrainingModule.train(
"modules/test/unittest/model/training_module/test.hdf5", "", "",
"modules/test/unittest/model/training_module/", 0.5)
def test_training_moreconv(mocked_config):
mocked_config.side_effect = TestTrainingModule.parse
TestTrainingModule.num_conv_layers = 2
TrainingModule.train(
"modules/test/unittest/model/training_module/test.hdf5", "", "",
"modules/test/unittest/model/training_module/", 0.5)
def test_training_no_valid_dataset(mocked_config):
mocked_config.side_effect = TestTrainingModule.parse
TrainingModule.train(
"modules/test/unittest/model/training_module/notvalid.hdf5", "",
"", "modules/test/unittest/model/training_module/", 0.5)
def test_training_loadednetwork(mocked_config):
mocked_config.side_effect = TestTrainingModule.parse
TrainingModule.train(
"modules/test/unittest/model/training_module/test.hdf5",
"modules/test/unittest/model/training_module/testnetwork.hdf5", "",
"modules/test/unittest/model/training_module/", 0.5)