def test_should_include_unknown_class_count_if_enabled(self):
with tf.Session() as session:
assert session.run(
calculate_sample_frequencies(
[[COLOR_1, COLOR_2, COLOR_3]], [COLOR_1],
use_unknown_class=True)) == [1.0, 2.0]
def test_should_return_total_count_for_multiple_all_matching_color(self):
with tf.Session() as session:
assert session.run(
calculate_sample_frequencies([[COLOR_1, COLOR_1, COLOR_1]],
[COLOR_1])) == [3.0]
def test_should_return_total_count_for_multiple_mixed_color(self):
with tf.Session() as session:
assert session.run(
calculate_sample_frequencies(
[[COLOR_1, COLOR_1, COLOR_2]],
[COLOR_1, COLOR_2])) == [2.0, 1.0]
def test_should_return_one_for_single_matching_color(self):
with tf.Session() as session:
assert session.run(
calculate_sample_frequencies([[COLOR_1]], [COLOR_1])) == [1.0]