def test_mutate(self):
dataf_a = dplyr.DataFrame(mtcars)
dataf_b = dataf_a.mutate(foo=1, bar=rl('gear+1'))
assert type(dataf_b) is dplyr.DataFrame
assert all(a == b
for a, b in zip(dataf_a.rx2('gear'), dataf_b.rx2('gear')))
assert all(a + 1 == b
for a, b in zip(dataf_a.rx2('gear'), dataf_b.rx2('bar')))
def test_join(self):
dataf_a = dplyr.DataFrame(mtcars)
dataf_b = dataf_a.mutate(foo=1)
dataf_c = dataf_a.inner_join(dataf_b, by=dataf_a.colnames)
all_names = list(dataf_a.colnames)
all_names.append('foo')
assert sorted(list(all_names)) == sorted(list(dataf_c.colnames))
assert tuple(all_names) == tuple(dataf_c.colnames)
def testJoin(self):
dataf_a = dplyr.DataFrame(mtcars)
dataf_b = dataf_a.mutate(foo=1)
dataf_c = dataf_a.inner_join(dataf_b)
all_names = list(dataf_a.colnames)
all_names.append('foo')
try:
# Python 3
self.assertCountEqual(all_names, dataf_c.colnames)
except AttributeError as ae:
# Python 2.7
self.assertItemsEqual(all_names, dataf_c.colnames)
def build_dataframe(monitor: Monitor) -> dplyr.DataFrame:
what = (
'susceptible',
'incubating',
'sick',
)
dataf = dplyr.DataFrame({
'what':
ro.StrVector([v for v in what for x in monitor.day]),
'day':
ro.IntVector([v for x in what for v in monitor.day]),
'count':
ro.IntVector([v for x in what for v in getattr(monitor, x)])
})
return dataf
def testFilter_OneFilter(self):
dataf = dplyr.DataFrame(mtcars)
ngear_gt_3 = len(tuple(x for x in dataf.rx2('gear') if x > 3))
dataf_filter = dataf.filter('gear > 3')
self.assertEqual(ngear_gt_3, dataf_filter.nrow)
def testSplitMergeFunction(self):
dataf = dplyr.DataFrame(mtcars)
dataf_by_gear = dataf.group_by('gear')
dataf_sum_gear = dataf_by_gear.summarize(foo='sum(gear)')
self.assertEquals(type(dataf_sum_gear), dplyr.DataFrame)
def test_collect(self):
dataf = dplyr.DataFrame(mtcars)
dataf_collected = dataf.collect()
# FIXME: no real test here. Just ensuring that it is returning
# without error
assert type(dataf_collected) is dplyr.DataFrame
def test_filter_onefilter_method(self):
dataf = dplyr.DataFrame(mtcars)
ngear_gt_3 = len(tuple(x for x in dataf.rx2('gear') if x > 3))
dataf_filter = dataf.filter(rl('gear > 3'))
assert ngear_gt_3 == dataf_filter.nrow
def test_filter_nofilter_function(self):
dataf = dplyr.DataFrame(mtcars)
dataf_filter = dplyr.filter(dataf)
assert dataf.nrow == dataf_filter.nrow
def test_distinct(self):
dataf_a = dplyr.DataFrame(mtcars)
res = dataf_a.distinct()
assert res.nrow == dataf_a.nrow
def test_group_by(self):
dataf_a = dplyr.DataFrame(mtcars)
dataf_g = dataf_a.group_by(rl('gear'))
assert dataf_g.is_grouped_df
assert not dataf_g.ungroup().is_grouped_df
assert dataf_g.is_grouped_df
def test_splitmerge_function(self):
dataf = dplyr.DataFrame(mtcars)
dataf_by_gear = dataf.group_by(rl('gear'))
dataf_avg_mpg = dataf_by_gear.summarize(foo=rl('mean(mpg)'))
assert isinstance(dataf_avg_mpg, dplyr.DataFrame)
def test_sample_select(self):
dataf_a = dplyr.DataFrame(mtcars)
res = dataf_a.select('gear')
assert res.nrow == dataf_a.nrow
def test_sample_frac(self):
dataf_a = dplyr.DataFrame(mtcars)
res = dataf_a.sample_frac(.5)
assert res.nrow == int(dataf_a.nrow / 2)
def test_sample_n(self):
dataf_a = dplyr.DataFrame(mtcars)
res = dataf_a.sample_n(5)
assert res.nrow == 5
def test_dataframe(self):
dataf = dplyr.DataFrame(mtcars)
# FIXME: no testing much at the moment...
assert isinstance(dataf, dplyr.DataFrame)
def test_splitmerge_function(self):
dataf = dplyr.DataFrame(mtcars)
dataf_by_gear = dataf.group_by('gear')
dataf_sum_gear = dataf_by_gear.summarize(foo='sum(gear)')
assert type(dataf_sum_gear) is dplyr.DataFrame
def test_filter_nofilter_method(self):
dataf = dplyr.DataFrame(mtcars)
dataf_filter = dataf.filter()
assert dataf.nrow == dataf_filter.nrow
def test_count(self):
dataf_a = dplyr.DataFrame(mtcars)
res = dataf_a.count()
assert tuple(res.rx2('n')) == (dataf_a.nrow, )
def test_filter_onefilter_function(self):
dataf = dplyr.DataFrame(mtcars)
ngear_gt_3 = len(tuple(x for x in dataf.rx2('gear') if x > 3))
dataf_filter = dplyr.filter(dataf, 'gear > 3')
assert ngear_gt_3 == dataf_filter.nrow
def test_mutate_at(self):
dataf_a = dplyr.DataFrame(mtcars)
dataf_b = dataf_a.mutate_at(StrVector(["gear"]), rl('sqrt'))
assert type(dataf_b) is dplyr.DataFrame
def test_mutate_all(self):
dataf_a = dplyr.DataFrame(mtcars)
dataf_b = dataf_a.mutate_all(rl('sqrt'))
assert type(dataf_b) is dplyr.DataFrame
def testFilter_NoFilter(self):
dataf = dplyr.DataFrame(mtcars)
dataf_filter = dataf.filter()
self.assertEqual(dataf.nrow, dataf_filter.nrow)
def test_arrange(self):
dataf = dplyr.DataFrame(mtcars)
dataf_arrange = dataf.arrange('mpg')
assert tuple(sorted(dataf.collect().rx2('mpg'))) == \
tuple(dataf_arrange.collect().rx2('mpg'))
def DataFrame(self, *args, **kwargs):
dataf = dplyr.DataFrame(*args, **kwargs)
res = dplyr.copy_to(self.dbcon, dataf, name=str(uuid.uuid4()))
print(res)
return res
