def testDeleteCubeSet2(self):
cubeSetName = "test-" + str(uuid.uuid4())
csvFilePath = cubeSetName + ".csv"
try:
shutil.copyfile("cubify/tests/testdata.csv", cubeSetName + ".csv")
except Exception:
shutil.copyfile("./testdata.csv", cubeSetName + ".csv")
binningFileName = "cubify/tests/test_binnings.json"
if os.path.isfile(binningFileName) == False:
binningFileName = "./test_binnings.json"
with open(binningFileName) as binnings_file:
binnings = json.load(binnings_file)
aggFileName = "cubify/tests/test_agg.json"
if os.path.isfile(aggFileName) == False:
aggFileName = "./test_agg.json"
with open(aggFileName) as agg_file:
aggs = json.load(agg_file)
cs = CubeSetService("testdb")
cs.createCubeSet("testOwner", cubeSetName, csvFilePath, binnings, aggs)
cs.deleteCubeSet(cubeSetName)
cubeSet = cs.getCubeSet(cubeSetName)
self.assertTrue(cubeSet == None)
os.remove(csvFilePath)
def testDeleteCubeSet(self):
cubeSetName = "test-" + str(uuid.uuid4())
csvFilePath = cubeSetName + ".csv"
try:
shutil.copyfile("cubify/tests/testdata.csv", cubeSetName + ".csv")
except Exception:
shutil.copyfile("./testdata.csv", cubeSetName + ".csv")
cs = CubeSetService("testdb")
cs.createCubeSet("testOwner", cubeSetName, csvFilePath, None, None)
cs.deleteCubeSet(cubeSetName)
cubeSet = cs.getCubeSet(cubeSetName)
self.assertTrue(cubeSet == None)
os.remove(csvFilePath)
def testCreateCubeSetWithAutoBinning(self):
cubeSetName = "test-" + str(uuid.uuid4())
csvFilePath = cubeSetName + ".csv"
try:
shutil.copyfile("cubify/tests/testdata.csv", cubeSetName + ".csv")
except Exception:
shutil.copyfile("./testdata.csv", cubeSetName + ".csv")
cs = CubeSetService("testdb")
cs.createCubeSet("testOwner", cubeSetName, csvFilePath)
cubeSet = cs.getCubeSet(cubeSetName)
self.assertTrue(cubeSet != None)
self.assertTrue(cubeSet["name"] == cubeSetName)
self.assertTrue(cubeSet["owner"] == "testOwner")
self.assertTrue(cubeSet["csvFilePath"] == csvFilePath)
self.assertTrue(cubeSet["sourceCube"] == cubeSetName + "_source")
self.assertTrue(cubeSet["binnedCube"] == cubeSetName + "_binned")
self.assertFalse("aggs" in cubeSet)
os.remove(csvFilePath)
def testCreateCubeSet(self):
cubeSetName = "test-" + str(uuid.uuid4())
csvFilePath = cubeSetName + ".csv"
try:
shutil.copyfile("cubify/tests/testdata.csv", cubeSetName + ".csv")
except Exception:
shutil.copyfile("./testdata.csv", cubeSetName + ".csv")
binningFileName = "cubify/tests/test_binnings.json"
if os.path.isfile(binningFileName) == False:
binningFileName = "./test_binnings.json"
with open(binningFileName) as binnings_file:
binnings = json.load(binnings_file)
aggFileName = "cubify/tests/test_agg.json"
if os.path.isfile(aggFileName) == False:
aggFileName = "./test_agg.json"
with open(aggFileName) as agg_file:
aggs = json.load(agg_file)
cs = CubeSetService("testdb")
cs.createCubeSet("testOwner", cubeSetName, csvFilePath, binnings, aggs)
cubeSet = cs.getCubeSet(cubeSetName)
self.assertTrue(cubeSet != None)
self.assertTrue(cubeSet["name"] == cubeSetName)
self.assertTrue(cubeSet["owner"] == "testOwner")
self.assertTrue(cubeSet["csvFilePath"] == csvFilePath)
self.assertTrue(cubeSet["sourceCube"] == cubeSetName + "_source")
self.assertTrue(cubeSet["binnedCube"] == cubeSetName + "_binned")
binnedCubeName = cubeSet["binnedCube"]
self.assertTrue(len(cubeSet["aggCubes"]) == 3)
self.assertTrue(binnedCubeName + "_" + aggs[0]["name"] in cubeSet["aggCubes"])
self.assertTrue(binnedCubeName + "_" + aggs[1]["name"] in cubeSet["aggCubes"])
self.assertTrue(binnedCubeName + "_" + aggs[2]["name"] in cubeSet["aggCubes"])
os.remove(csvFilePath)
