class TestSum(unittest.TestCase): """Automatic tests""" def setUp(self): self.table_mng = TableManager() self.table_mng.table_df.from_dict(ROWS) def test_write_json(self): """ Test if the file is created properly.When the path is passed as parameter to pands.to_json, the return should be None """ result = self.table_mng.write_table_json() self.assertIsNone(result, "Should be a None") self.assertEqual(True, os.path.exists(r"../app/integrations/data.json")) def test_dataframe_columns(self): """Test integrity of columns""" self.table_mng.create_table(["Oceania", "Americas"]) self.assertIn( "country", self.table_mng.table_df.columns, "Country should be a column in df", ) self.assertEqual(self.table_mng.table_df.shape[1], 4, "Should be 4 columns") self.assertEqual( self.table_mng.table_df.columns.values.tolist(), ["region", "country", "language", "time"], )
def main(): mongodb = Mongo() table_mng = TableManager() auth = Auth() regions = get_regions(os.getenv("REGIONS_API")) table_mng.create_table(regions) auth.get_authorization() if not auth.is_authenticated: logging.info("User is not authenticated so database can't be shown") return mongodb.insert_documents(table_mng.table_df) mongodb.list_documents()