QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-12 12:47:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Sample test data (#10487)

Browse Source 
Co-authored-by: bogdan kyryliuk <bogdankyryliuk@dropbox.com>
This commit is contained in:
Bogdan
2020-08-03 09:08:49 -07:00
committed by GitHub
parent 821916a681
commit ab404ea2cf
8 changed files with 46 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
superset/examples/unicode_test_data.py
View File

@@ -36,7 +36,9 @@ from .helpers import (
)
def load_unicode_test_data(only_metadata: bool = False, force: bool = False) -> None:
def load_unicode_test_data(
only_metadata: bool = False, force: bool = False, sample: bool = False
) -> None:
"""Loading unicode test dataset from a csv file in the repo"""
tbl_name = "unicode_test"
database = utils.get_example_database()
@@ -50,6 +52,7 @@ def load_unicode_test_data(only_metadata: bool = False, force: bool = False) ->
# generate date/numeric data
df["dttm"] = datetime.datetime.now().date()
df["value"] = [random.randint(1, 100) for _ in range(len(df))]
df = df.head(100) if sample else df
df.to_sql( # pylint: disable=no-member
tbl_name,
database.get_sqla_engine(),