def test_lm_all_variants_no_load(self):
for variant in [
MODEL_VARIANT_01_ORIGINAL, MODEL_VARIANT_02_EDITED,
MODEL_VARIANT_03_CAPTURED, MODEL_VARIANT_04_CORRECTED,
MODEL_VARIANT_05_SCALED33
]:
load_sd15ch1_models(data_home=self.tmpdir,
load_images=False,
variant=variant,
color=False)
def test_lm_return_X_y_withnone(self):
model_data = load_sd15ch1_models(data_home=self.tmpdir,
load_images=False,
with_model_ids=False,
with_modeltype_ids=False,
return_X_y=True)
self.assertEqual(len(model_data), 1)
def test_lm_no_load(self):
model_data = load_sd15ch1_models(data_home=self.tmpdir,
download_if_missing=True,
load_images=False,
variant=MODEL_VARIANT_05_SCALED33,
color=False,
with_model_ids=True,
with_modeltype_ids=True,
return_X_y=False)
for k, v in six.iteritems(model_data):
print(k, v)