def setup_class(cls): TestBase.setup_class() cls.ctx, cls.expt, cls.protocol = cls.make_experiment_fixture() print("Importing example field data: {}".format(EXAMPLE_FIELD_DATA_CSV)) import_csv(cls.ctx, container_uri=str(cls.expt.getURI().toString()), protocol_uri=str(cls.protocol.getURI().toString()), files=[EXAMPLE_FIELD_DATA_CSV])
def should_use_existing_sources(self): expt2 = self.ctx.insertProject("project2","project2",DateTime()).insertExperiment("purpose", DateTime()) protocol2 = self.ctx.insertProtocol("protocol", "description") import_csv(self.ctx, container_uri=expt2.getURI().toString(), protocol_uri=protocol2.getURI().toString(), files=[EXAMPLE_FIELD_DATA_CSV]) expected_source_names = np.unique(self.df.Site) sources = self.ctx.getTopLevelSources() assert_equals(len(expected_source_names), len(list(iterable(sources))))
def should_use_existing_site_epoch_when_present(self): expt2 = self.ctx.insertProject("project2","project2",DateTime()).insertExperiment("purpose", DateTime()) protocol2 = self.ctx.insertProtocol("protocol", "description") import_csv(self.ctx, container_uri=expt2.getURI().toString(), protocol_uri=protocol2.getURI().toString(), files=[EXAMPLE_FIELD_DATA_CSV]) import_csv(self.ctx, container_uri=expt2.getURI().toString(), protocol_uri=protocol2.getURI().toString(), files=[EXAMPLE_FIELD_DATA_CSV]) self.check_epoch_per_site(expt2)
def import_wrapper(ctx, container=None, protocol=None, files=None, timezone=None, header_rows=None, date_column=None, measurement_column=None, **args): return import_csv(ctx, container_uri=container, protocol_uri=protocol, files=files, timezone=timezone, csv_header_row=header_rows, date_column=date_column, first_measurement_column=measurement_column)