Skip to content

Commit 90002a7

Browse files
author
Michael Erickson
committed
Updated casout to include name
1 parent 3a515b1 commit 90002a7

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

swat/tests/cas/test_table.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -872,7 +872,7 @@ def test_drop_duplicates(self):
872872
tbl = self.table
873873
df = self.get_cars_df
874874
# drop duplicates for single subset
875-
tbl_dropped = tbl.drop_duplicates(casout={'replace': True}, subset='Make')
875+
tbl_dropped = tbl.drop_duplicates(casout={'replace': True, 'name':'drop-test-1'}, subset='Make')
876876
df_dropped = df.drop_duplicates(subset='Make')
877877

878878
# Equivalent to pandas in size
@@ -882,7 +882,7 @@ def test_drop_duplicates(self):
882882
self.assertEquals(tbl_dropped['Make'].nunique(), len(tbl_dropped))
883883

884884
# drop duplicates for multi-element subset
885-
tbl_dropped_multi = tbl.drop_duplicates(casout={'replace': True},
885+
tbl_dropped_multi = tbl.drop_duplicates(casout={'replace': True, 'name':'drop-test-2'},
886886
subset=['Country', 'Type'])
887887
df_dropped_multi = df.drop_duplicates(subset=['Country', 'Type'])
888888

@@ -900,14 +900,14 @@ def test_drop_duplicates(self):
900900
{'name': 'Origin', 'type': 'CHAR', 'length': 6},
901901
{'name': 'DriveTrain', 'type': 'CHAR', 'length': 5}
902902
]}
903-
subset = self.s.upload_frame(fetchTable, casout={'replace': True},
903+
subset = self.s.upload_frame(fetchTable, casout={'replace': True, 'name':'drop-test-3'},
904904
importOptions=importOptions)
905905

906906
# This table is like tbl, but with nDuplicate fully duplicate rows
907907
duplicate_table = tbl.append(subset)
908908

909909
# Drop duplicates without subset (checks all cols)
910-
tbl_dropped_all = duplicate_table.drop_duplicates(casout={'replace': True})
910+
tbl_dropped_all = duplicate_table.drop_duplicates(casout={'replace': True, 'name':'drop-test-4'})
911911

912912
# Make sure that the correct amount of rows were dropped
913913
self.assertEquals(len(tbl), len(tbl_dropped_all))

0 commit comments

Comments
 (0)