def setUpClass(cls): print(cls.__doc__) _, cls.engine = start_postgre_engine('test', False) cls.conn = cls.engine.connect() cls.paths = return_files_paths() cls.dataset = return_dataset(cls.paths[0]) cls.session = dbProxy.create_session(cls.engine)
def setUp(self): self.test_length = self.TEST_LENGTH self.session = dbProxy.create_session(self.engine) # insert some rows and return their pks samples = util_populate_table(self.dataset, self.test_length) # pick a random sample self.i, self.test_point_pk, self.test_areas_pk, self.long, self.lat = pick_random_sample( self.dataset, samples)
def setUp(self): self.test_length = self.TEST_LENGTH self.session = dbProxy.create_session(self.engine) # insert some rows and return their pks samples = util_populate_table(self.dataset, self.test_length) # pick a random sample self.i, self.test_point_pk, self.test_areas_pk, self.long, self.lat = pick_random_sample( self.dataset, samples )
def test_bulk_dump(self): """Test Xco2.bulk_dump()""" from src.formatdata import bulk_dump print('#### TEST5 ####') session2 = dbProxy.create_session(db='test', engine=self.engine) util_truncate_table(session2, [Xco2, Areas]) bulk_dump(create_generator_from_dataset(self.dataset, 8)) rows = self.session.query(Xco2).count() try: self.assertEqual(rows, 8) print('PASSED') except AssertionError: print('FAILED')
def test_bulk_dump(self): """Test Xco2.bulk_dump()""" from src.formatdata import bulk_dump print('#### TEST5 ####') session2 = dbProxy.create_session(db='test', engine=self.engine) util_truncate_table(session2, [Xco2, Areas]) bulk_dump( create_generator_from_dataset(self.dataset, 8) ) rows = self.session.query(Xco2).count() try: self.assertEqual(rows, 8) print('PASSED') except AssertionError: print('FAILED')
def setUp(self): self.test_length = self.TEST_LENGTH self.session = dbProxy.create_session(db='test', engine=self.engine) self.samples = util_populate_table(self.dataset, self.test_length)